จัดกลุ่มตามชื่อเขตข้อมูลหลายชื่อใน java 8


95

ฉันพบรหัสสำหรับจัดกลุ่มวัตถุตามชื่อเขตข้อมูลจาก POJO ด้านล่างนี้คือรหัสสำหรับสิ่งนั้น:

public class Temp {

    static class Person {

        private String name;
        private int age;
        private long salary;

        Person(String name, int age, long salary) {

            this.name = name;
            this.age = age;
            this.salary = salary;
        }

        @Override
        public String toString() {
            return String.format("Person{name='%s', age=%d, salary=%d}", name, age, salary);
        }
    }

    public static void main(String[] args) {
        Stream<Person> people = Stream.of(new Person("Paul", 24, 20000),
                new Person("Mark", 30, 30000),
                new Person("Will", 28, 28000),
                new Person("William", 28, 28000));
        Map<Integer, List<Person>> peopleByAge;
        peopleByAge = people
                .collect(Collectors.groupingBy(p -> p.age, Collectors.mapping((Person p) -> p, toList())));
        System.out.println(peopleByAge);
    }
}

และผลลัพธ์คือ (ซึ่งถูกต้อง):

{24=[Person{name='Paul', age=24, salary=20000}], 28=[Person{name='Will', age=28, salary=28000}, Person{name='William', age=28, salary=28000}], 30=[Person{name='Mark', age=30, salary=30000}]}

แต่ถ้าฉันต้องการจัดกลุ่มตามหลายเขตข้อมูลล่ะ? เห็นได้ชัดว่าฉันสามารถส่ง POJO ในgroupingBy()วิธีการบางอย่างได้หลังจากใช้equals()เมธอดใน POJO นั้น แต่มีตัวเลือกอื่นเช่นฉันสามารถจัดกลุ่มตามฟิลด์มากกว่าหนึ่งฟิลด์จาก POJO ที่กำหนดได้หรือไม่

เช่นในกรณีของฉันฉันต้องการจัดกลุ่มตามชื่อและอายุ


1
เคล็ดลับคือเพียงสร้างสตริงที่ไม่ซ้ำกันจากทุกฟิลด์
Marko Topolnik

3
BTW mappingในฐานะตัวรวบรวมดาวน์สตรีมซ้ำซ้อนในโค้ดที่คุณโพสต์
Marko Topolnik

8
people.collect(groupingBy(p -> Arrays.asList(p.name, p.age)))วิธีการแก้ปัญหาได้อย่างรวดเร็วและสกปรก
Misha

คำตอบ:


170

คุณมีทางเลือกสองสามทางที่นี่ วิธีที่ง่ายที่สุดคือการล่ามโซ่นักสะสมของคุณ:

Map<String, Map<Integer, List<Person>>> map = people
    .collect(Collectors.groupingBy(Person::getName,
        Collectors.groupingBy(Person::getAge));

จากนั้นเพื่อรับรายชื่อคนอายุ 18 ปีที่เรียกว่า Fred คุณจะใช้:

map.get("Fred").get(18);

ตัวเลือกที่สองคือการกำหนดคลาสที่แสดงถึงการจัดกลุ่ม สิ่งนี้สามารถอยู่ในตัวบุคคล รหัสนี้ใช้recordแต่อาจเป็นคลาส (ที่มีequalsและhashCodeกำหนด) ในเวอร์ชันของ Java ก่อนที่จะเพิ่ม JEP 359:

class Person {
    record NameAge(String name, int age) { }

    public NameAge getNameAge() {
        return new NameAge(name, age);
    }
}

จากนั้นคุณสามารถใช้:

Map<NameAge, List<Person>> map = people.collect(Collectors.groupingBy(Person::getNameAge));

และค้นหาด้วย

map.get(new NameAge("Fred", 18));

ในที่สุดหากคุณไม่ต้องการใช้บันทึกกลุ่มของคุณเองกรอบงาน Java จำนวนมากรอบ ๆ จะมีpairคลาสที่ออกแบบมาสำหรับสิ่งประเภทนี้ ตัวอย่างเช่นapache commons pairหากคุณใช้หนึ่งในไลบรารีเหล่านี้คุณสามารถกำหนดคีย์ให้กับแมปคู่ของชื่อและอายุ:

Map<Pair<String, Integer>, List<Person>> map =
    people.collect(Collectors.groupingBy(p -> Pair.of(p.getName(), p.getAge())));

และดึงข้อมูลด้วย:

map.get(Pair.of("Fred", 18));

โดยส่วนตัวแล้วฉันไม่เห็นคุณค่ามากนักใน tuples ทั่วไปในตอนนี้ที่ระเบียนมีให้บริการในภาษาเนื่องจากระเบียนแสดงเจตนาได้ดีขึ้นและต้องการรหัสน้อยมาก


5
Function<T,U>ยังซ่อนเจตนาในแง่นี้ - แต่คุณจะไม่เห็นใครประกาศอินเทอร์เฟซการทำงานของตนเองสำหรับแต่ละขั้นตอนการทำแผนที่ เจตนามีอยู่แล้วในร่างกายแลมด้า เช่นเดียวกับสิ่งทอ: เหมาะสำหรับประเภทกาวระหว่างส่วนประกอบ API ชั้นเรียนกรณีของ BTW Scala คือ IMHO เป็นชัยชนะครั้งใหญ่ในแง่ของความกระชับและการเปิดเผยเจตนา
Marko Topolnik

1
ใช่ฉันเห็นประเด็นของคุณ ฉันเดาว่า (เช่นเคย) ขึ้นอยู่กับวิธีการใช้ ตัวอย่างที่ฉันให้ไว้ข้างต้น - การใช้คู่เป็นกุญแจสำคัญของแผนที่เป็นตัวอย่างที่ดีในการไม่ทำ ฉันไม่ค่อยคุ้นเคยกับสกาล่า - ฉันจะต้องเริ่มเรียนรู้มันเมื่อฉันได้ยินสิ่งดีๆ
ปรินเตอร์

1
แค่คิดความสามารถในการประกาศNameAgeเป็นหนึ่งซับ: case class NameAge { val name: String; val age: Int }--- และคุณได้รับequals, hashCodeและtoString!
Marko Topolnik

1
Nice - รายการอื่นถูกส่งไปยังคิว 'must do' ของฉัน เป็น FIFO น่าเสียดาย!
ปรินเตอร์

@sprinter ประเภทในข้อมูลโค้ดแรกไม่ถูกต้องและควรเปลี่ยนเป็นMap<String, Map<Integer, List<Person>>> map
kasur

39

ดูรหัสที่นี่:

คุณสามารถสร้างฟังก์ชันและปล่อยให้มันทำงานแทนคุณได้แบบ Functional Style!

Function<Person, List<Object>> compositeKey = personRecord ->
    Arrays.<Object>asList(personRecord.getName(), personRecord.getAge());

ตอนนี้คุณสามารถใช้เป็นแผนที่:

Map<Object, List<Person>> map =
people.collect(Collectors.groupingBy(compositeKey, Collectors.toList()));

ไชโย!


2
ฉันใช้วิธีนี้ แต่แตกต่างกัน ฟังก์ชัน <Person, String> compositeKey = personRecord -> StringUtils.join (personRecord.getName (), personRecord.getAge ());
bpedroso

8

groupingByวิธีการได้พารามิเตอร์แรกคือFunction<T,K>ที่:

@param <T>ประเภทขององค์ประกอบอินพุต

@param <K>ประเภทของคีย์

หากเราแทนที่แลมบ์ดาด้วยคลาสที่ไม่ระบุตัวตนในโค้ดของคุณเราจะเห็นบางประเภท:

people.stream().collect(Collectors.groupingBy(new Function<Person, int>() {
            @Override
            public int apply(Person person) {
                return person.getAge();
            }
        }));

<K>เพียงแค่ตอนนี้เปลี่ยนพารามิเตอร์ขาออก ในกรณีนี้ฉันใช้คลาสคู่จาก org.apache.commons.lang3.tuple สำหรับการจัดกลุ่มตามชื่อและอายุ แต่คุณสามารถสร้างคลาสของคุณเองเพื่อกรองกลุ่มได้ตามต้องการ

people.stream().collect(Collectors.groupingBy(new Function<Person, Pair<Integer, String>>() {
                @Override
                public YourFilter apply(Person person) {
                    return Pair.of(person.getAge(), person.getName());
                }
            }));

สุดท้ายหลังจากแทนที่ด้วย lambda back โค้ดจะมีลักษณะดังนี้:

Map<Pair<Integer,String>, List<Person>> peopleByAgeAndName = people.collect(Collectors.groupingBy(p -> Pair.of(person.getAge(), person.getName()), Collectors.mapping((Person p) -> p, toList())));

ใช้แล้วได้List<String>อะไร?
Alex78191

7

สวัสดีคุณสามารถเชื่อมของคุณgroupingByKeyเช่น

Map<String, List<Person>> peopleBySomeKey = people
                .collect(Collectors.groupingBy(p -> getGroupingByKey(p), Collectors.mapping((Person p) -> p, toList())));



//write getGroupingByKey() function
private String getGroupingByKey(Person p){
return p.getAge()+"-"+p.getName();
}

2

กำหนดคลาสสำหรับนิยามคีย์ในกลุ่มของคุณ

class KeyObj {

    ArrayList<Object> keys;

    public KeyObj( Object... objs ) {
        keys = new ArrayList<Object>();

        for (int i = 0; i < objs.length; i++) {
            keys.add( objs[i] );
        }
    }

    // Add appropriate isEqual() ... you IDE should generate this

}

ตอนนี้อยู่ในรหัสของคุณ

peopleByManyParams = people
            .collect(Collectors.groupingBy(p -> new KeyObj( p.age, p.other1, p.other2 ), Collectors.mapping((Person p) -> p, toList())));

3
นั่นเป็นเพียงการคิดค้นสิ่งใหม่Ararys.asList()- ซึ่ง BTW เป็นตัวเลือกที่ดีสำหรับเคสของ OP
Marko Topolnik

และยังคล้ายกับPairตัวอย่างที่กล่าวถึงในอีกตัวอย่าง แต่ไม่ จำกัด อาร์กิวเมนต์
Benny Bottema

นอกจากนี้คุณต้องทำให้สิ่งนี้ไม่เปลี่ยนรูป (และคำนวณhashCode) ครั้งเดียว)
RobAu

2

คุณสามารถใช้ List เป็นตัวแยกประเภทสำหรับหลาย ๆ ฟิลด์ แต่คุณต้องรวมค่า null เป็นตัวเลือก

Function<String, List> classifier = (item) -> List.of(
    item.getFieldA(),
    item.getFieldB(),
    Optional.ofNullable(item.getFieldC())
);

Map<List, List<Item>> grouped = items.stream()
    .collect(Collectors.groupingBy(classifier));

1

ฉันต้องการทำรายงานสำหรับ บริษัท จัดเลี้ยงที่ให้บริการอาหารกลางวันสำหรับลูกค้าหลายราย กล่าวอีกนัยหนึ่งการจัดเลี้ยงอาจมีใน บริษัท หรือมากกว่านั้นที่รับคำสั่งจากการจัดเลี้ยงและต้องรู้ว่าต้องผลิตอาหารกลางวันกี่มื้อต่อวันสำหรับลูกค้าทั้งหมด!

เพียงสังเกตว่าฉันไม่ได้ใช้การเรียงลำดับเพื่อไม่ให้ตัวอย่างนี้ซับซ้อนเกินไป

นี่คือรหัสของฉัน:

@Test
public void test_2() throws Exception {
    Firm catering = DS.firm().get(1);
    LocalDateTime ldtFrom = LocalDateTime.of(2017, Month.JANUARY, 1, 0, 0);
    LocalDateTime ldtTo = LocalDateTime.of(2017, Month.MAY, 2, 0, 0);
    Date dFrom = Date.from(ldtFrom.atZone(ZoneId.systemDefault()).toInstant());
    Date dTo = Date.from(ldtTo.atZone(ZoneId.systemDefault()).toInstant());

    List<PersonOrders> LON = DS.firm().getAllOrders(catering, dFrom, dTo, false);
    Map<Object, Long> M = LON.stream().collect(
            Collectors.groupingBy(p
                    -> Arrays.asList(p.getDatum(), p.getPerson().getIdfirm(), p.getIdProduct()),
                    Collectors.counting()));

    for (Map.Entry<Object, Long> e : M.entrySet()) {
        Object key = e.getKey();
        Long value = e.getValue();
        System.err.println(String.format("Client firm :%s, total: %d", key, value));
    }
}

0

นี่คือวิธีที่ฉันจัดกลุ่มตาม branchCode และ prdId หลายฟิลด์เพียงแค่โพสต์สำหรับคนที่ต้องการ

    import java.math.BigDecimal;
    import java.math.BigInteger;
    import java.util.ArrayList;
    import java.util.Iterator;
    import java.util.LinkedList;
    import java.util.List;
    import java.util.Map;
    import java.util.stream.Collectors;

    /**
     *
     * @author charudatta.joshi
     */
    public class Product1 {

        public BigInteger branchCode;
        public BigInteger prdId;
        public String accountCode;
        public BigDecimal actualBalance;
        public BigDecimal sumActBal;
        public BigInteger countOfAccts;

        public Product1() {
        }

        public Product1(BigInteger branchCode, BigInteger prdId, String accountCode, BigDecimal actualBalance) {
            this.branchCode = branchCode;
            this.prdId = prdId;
            this.accountCode = accountCode;
            this.actualBalance = actualBalance;
        }

        public BigInteger getCountOfAccts() {
            return countOfAccts;
        }

        public void setCountOfAccts(BigInteger countOfAccts) {
            this.countOfAccts = countOfAccts;
        }

        public BigDecimal getSumActBal() {
            return sumActBal;
        }

        public void setSumActBal(BigDecimal sumActBal) {
            this.sumActBal = sumActBal;
        }

        public BigInteger getBranchCode() {
            return branchCode;
        }

        public void setBranchCode(BigInteger branchCode) {
            this.branchCode = branchCode;
        }

        public BigInteger getPrdId() {
            return prdId;
        }

        public void setPrdId(BigInteger prdId) {
            this.prdId = prdId;
        }

        public String getAccountCode() {
            return accountCode;
        }

        public void setAccountCode(String accountCode) {
            this.accountCode = accountCode;
        }

        public BigDecimal getActualBalance() {
            return actualBalance;
        }

        public void setActualBalance(BigDecimal actualBalance) {
            this.actualBalance = actualBalance;
        }

        @Override
        public String toString() {
            return "Product{" + "branchCode:" + branchCode + ", prdId:" + prdId + ", accountCode:" + accountCode + ", actualBalance:" + actualBalance + ", sumActBal:" + sumActBal + ", countOfAccts:" + countOfAccts + '}';
        }

        public static void main(String[] args) {
            List<Product1> al = new ArrayList<Product1>();
            System.out.println(al);
            al.add(new Product1(new BigInteger("01"), new BigInteger("11"), "001", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("01"), new BigInteger("11"), "002", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("01"), new BigInteger("12"), "003", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("01"), new BigInteger("12"), "004", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("01"), new BigInteger("12"), "005", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("01"), new BigInteger("13"), "006", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("02"), new BigInteger("11"), "007", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("02"), new BigInteger("11"), "008", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("02"), new BigInteger("12"), "009", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("02"), new BigInteger("12"), "010", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("02"), new BigInteger("12"), "011", new BigDecimal("10")));
            al.add(new Product1(new BigInteger("02"), new BigInteger("13"), "012", new BigDecimal("10")));
            //Map<BigInteger, Long> counting = al.stream().collect(Collectors.groupingBy(Product1::getBranchCode, Collectors.counting()));
            // System.out.println(counting);

            //group by branch code
            Map<BigInteger, List<Product1>> groupByBrCd = al.stream().collect(Collectors.groupingBy(Product1::getBranchCode, Collectors.toList()));
            System.out.println("\n\n\n" + groupByBrCd);

             Map<BigInteger, List<Product1>> groupByPrId = null;
              // Create a final List to show for output containing one element of each group
            List<Product> finalOutputList = new LinkedList<Product>();
            Product newPrd = null;
            // Iterate over resultant  Map Of List
            Iterator<BigInteger> brItr = groupByBrCd.keySet().iterator();
            Iterator<BigInteger> prdidItr = null;    



            BigInteger brCode = null;
            BigInteger prdId = null;

            Map<BigInteger, List<Product>> tempMap = null;
            List<Product1> accListPerBr = null;
            List<Product1> accListPerBrPerPrd = null;

            Product1 tempPrd = null;
            Double sum = null;
            while (brItr.hasNext()) {
                brCode = brItr.next();
                //get  list per branch
                accListPerBr = groupByBrCd.get(brCode);

                // group by br wise product wise
                groupByPrId=accListPerBr.stream().collect(Collectors.groupingBy(Product1::getPrdId, Collectors.toList()));

                System.out.println("====================");
                System.out.println(groupByPrId);

                prdidItr = groupByPrId.keySet().iterator();
                while(prdidItr.hasNext()){
                    prdId=prdidItr.next();
                    // get list per brcode+product code
                    accListPerBrPerPrd=groupByPrId.get(prdId);
                    newPrd = new Product();
                     // Extract zeroth element to put in Output List to represent this group
                    tempPrd = accListPerBrPerPrd.get(0);
                    newPrd.setBranchCode(tempPrd.getBranchCode());
                    newPrd.setPrdId(tempPrd.getPrdId());

                    //Set accCOunt by using size of list of our group
                    newPrd.setCountOfAccts(BigInteger.valueOf(accListPerBrPerPrd.size()));
                    //Sum actual balance of our  of list of our group 
                    sum = accListPerBrPerPrd.stream().filter(o -> o.getActualBalance() != null).mapToDouble(o -> o.getActualBalance().doubleValue()).sum();
                    newPrd.setSumActBal(BigDecimal.valueOf(sum));
                    // Add product element in final output list

                    finalOutputList.add(newPrd);

                }

            }

            System.out.println("+++++++++++++++++++++++");
            System.out.println(finalOutputList);

        }
    }

ผลลัพธ์มีดังนี้:

+++++++++++++++++++++++
[Product{branchCode:1, prdId:11, accountCode:null, actualBalance:null, sumActBal:20.0, countOfAccts:2}, Product{branchCode:1, prdId:12, accountCode:null, actualBalance:null, sumActBal:30.0, countOfAccts:3}, Product{branchCode:1, prdId:13, accountCode:null, actualBalance:null, sumActBal:10.0, countOfAccts:1}, Product{branchCode:2, prdId:11, accountCode:null, actualBalance:null, sumActBal:20.0, countOfAccts:2}, Product{branchCode:2, prdId:12, accountCode:null, actualBalance:null, sumActBal:30.0, countOfAccts:3}, Product{branchCode:2, prdId:13, accountCode:null, actualBalance:null, sumActBal:10.0, countOfAccts:1}]

หลังจากฟอร์แมตแล้ว:

[
Product{branchCode:1, prdId:11, accountCode:null, actualBalance:null, sumActBal:20.0, countOfAccts:2}, 
Product{branchCode:1, prdId:12, accountCode:null, actualBalance:null, sumActBal:30.0, countOfAccts:3}, 
Product{branchCode:1, prdId:13, accountCode:null, actualBalance:null, sumActBal:10.0, countOfAccts:1}, 
Product{branchCode:2, prdId:11, accountCode:null, actualBalance:null, sumActBal:20.0, countOfAccts:2}, 
Product{branchCode:2, prdId:12, accountCode:null, actualBalance:null, sumActBal:30.0, countOfAccts:3}, 
Product{branchCode:2, prdId:13, accountCode:null, actualBalance:null, sumActBal:10.0, countOfAccts:1}
]
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.