Java中如何用Stream分组并求各组数量寻技术

比如给了我一个班级的所有人的姓氏和姓名对象集合，根据这个集合求出各姓氏有多少人。直接上代码。

User的实体类对象为

@Data
@NoArgsConstructor
@AllArgsConstructor
public class User {
    private String firstName; // 姓氏
    private String name; // 姓名
}

List<User> userList = new ArrayList<>();
userList.add(new User("张","张三"));
userList.add(new User("张","张四"));
userList.add(new User("张","张五"));
userList.add(new User("李","李一"));
userList.add(new User("李","李二"));
Map<String, Integer> groupCount = userList.stream()
        .collect(Collectors.toMap(User::getFirstName, el -> 1, Integer::sum));
System.out.println(groupCount);

输出的结果为：

即姓张的有3个，姓李的有两个。

详解：将该集合的stream流用Collectors对象转成Map，用User对象的firstName作为Key，默认键值为1，在处理键冲突的函数中，将上一次的值与默认值1相加。比如处理第一个姓张的人的时候，默认值为1，处理第二个也是姓张的人的时候默认值也是1，将1和1相加得2，（即有两个姓张的人了），处理第三个姓张的人的时候，将上次的结果2与第三个姓张返回的默认值1相加，得3，以此类推...得到处理结果。

附：Java中用Stream分组并求各组数量

一：多字段分组统计

方法一

public static void main(String[] args) {
       List<Test> list = new ArrayList<>();
        for(int i = 0; i< 5; i++){
            Test test = new Test();
            //主键id
            test.setId(Long.valueOf(i));
            //金额
            test.setMoney(new BigDecimal(i));
            //数量
            test.setCount(new BigDecimal(i));
            list.add(test);
        }
        for(int i = 0; i< 2; i++){
            Test test = new Test();
            //主键id
            test.setId(Long.valueOf(i));
            //金额
            test.setMoney(new BigDecimal(i));
            //数量
            test.setCount(new BigDecimal(i));
            list.add(test);
        }
        List<Test> resultList = new ArrayList<>();
        //先根据id分组
        Map<Long,List<Test>> map = list.stream().collect(Collectors.groupingBy(m->m.getId()));
        //分组后求 总金额，总数量
        map.keySet().stream().forEach(key->{
            Test info = new Test();
            List<Test> listInfo = map.get(key);
            //总金额
            BigDecimal sumMoney = listInfo.stream().map(Test::getMoney).reduce(BigDecimal.ZERO,BigDecimal::add);
            //总数量
            BigDecimal sumCount = listInfo.stream().map(Test::getCount).reduce(BigDecimal.ZERO,BigDecimal::add);
            info.setId(key);
            info.setMoney(sumMoney);
            info.setCount(sumCount);
            resultList.add(info);
        });
        System.out.println(resultList);
    }

方法二

List<Student> collect = stus.stream().collect(Collectors.toMap(k ->k.getId()+k.getBj(), e -> e, (o1, o2) -> {
            o1.setK0qmkcje((null==o1.getK0qmkcje()?BigDecimal.ZERO:o1.getK0qmkcje()).add((null==o2.getK0qmkcje()?BigDecimal.ZERO:o2.getK0qmkcje())).setScale(0,BigDecimal.ROUND_HALF_UP));
            return o1;
        })).values().stream().collect(Collectors.toList());

获取此类格式数据 Map<String, BigDecimal>

public static void main(String[] args) {
   Map<String, BigDecimal> ageGroup = list.stream().collect(Collectors.groupingBy(Student::getName
            , Collectors.collectingAndThen(Collectors.toList()
                    , x -> x.stream().map(Student::getStature).reduce(BigDecimal.ZERO, BigDecimal::add))));
}

public static void main(String[] args) {
    Map<String, BigDecimal> ageGroup = list.stream().collect(Collectors.groupingBy(Student::getName
            , MyCollector.summingDecimal(Student::getStature)));
    System.out.println(ageGroup);
}

总结

本文地址：https://www.xunjs.com/java/view_19192.html

Tags：数量 java Stream

相关推荐