Java Stream对List集合做分组统计

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

写代码时如果不想通过SQL查询做分组统计，通过Stream流也是可以做到的，现用简单demo记录一下。

先创建一个学生类

import lombok.AllArgsConstructor;
import lombok.Data;
 
@Data
@AllArgsConstructor
public class Student {
    private Integer id;
 
    private String name;
 
    private String sex;
 
    private Integer age;
 
    private Integer classNumber;
}

然后造几个数据，放到一个List里面

import java.util.ArrayList;
import java.util.List;
 
public class Test {
    public static void main(String[] args) {
        List<Student> studentList = new ArrayList<Student>();
        studentList.add(new Student(1, "张三", "男", 20, 2));
        studentList.add(new Student(2, "李四", "男", 21, 1));
        studentList.add(new Student(3, "王五", "女", 19, 1));
        studentList.add(new Student(4, "赵六", "男", 19, 1));
        studentList.add(new Student(5, "王大锤", "男", 20, 2));
}

1.如果需要对学生集合按班级分组，则可以这样写

Map<Integer, List<Student>> studentGroup = studentList.stream().collect(Collectors.groupingBy(Student::getClassNumber));
        for(Map.Entry<Integer, List<Student>> entry : studentGroup.entrySet()){
            System.out.println("key ："+entry.getKey()+", value ："+entry.getValue());
        }

运行结果：

key ：1, value ：[Student(id=2, name=李四, sex=男, age=21, classNumber=1), Student(id=3, name=王五, sex=女, age=19, classNumber=1), Student(id=4, name=赵六, sex=男, age=19, classNumber=1)]
key ：2, value ：[Student(id=1, name=张三, sex=男, age=20, classNumber=2), Student(id=5, name=王大锤, sex=男, age=20, classNumber=2)]

2.如果是对学生按班级分组，并统计每个班的人数

Map<Integer, Long> map = studentList.stream().collect(Collectors.groupingBy(Student::getClassNumber, Collectors.counting()));
        for(Map.Entry<Integer, Long> entry : map.entrySet()){
            System.out.println("key ："+entry.getKey()+", value ："+entry.getValue());
        }

运行结果

key ：1, value ：3
key ：2, value ：2