Java Stream流 Map 分组方式汇总

「爱情、让人受尽委屈。」 2024-03-25 22:12 21阅读 0赞

java老式的分组方式(对list for循环然后 if判断 放入map) 代码复杂,易读性差,维护性差,故本文汇总了Stream流中的分组方法供大家参考,如对您有帮助,请抬抬高贵的小手点个赞吧,欢迎大佬留下高见

(以下方法默认都是java8的方法,java9新增方法有标注)

List<Student> studentList = Arrays.asList(
    
    new Student(1L,"小红","篮球","红色",22),
    
    new Student(2L,"小白","足球","蓝色",11),
    
    new Student(3L,"小牛","足球","黑色",33),
    
    new Student(4L,"小羊","足球","红色",17),
    
    new Student(5L,"小绿","足球","蓝色",24));

### 1.常规按对象的字段直接将对象分组 ###

按颜色分组

Map<String, List<Student>> byColor = studentList.stream()
    .collect(groupingBy(Student::getColor));

打印Map结果：

\{红色=\[Student\{id=1, name='小红', hobby='篮球', color='红色', age=22\}, Student\{id=4, name='小羊', hobby='足球', color='红色', age=17\}\],

蓝色=\[Student\{id=2, name='小白', hobby='足球', color='蓝色', age=11\}, Student\{id=5, name='小绿', hobby='足球', color='蓝色', age=24\}\],

黑色=\[Student\{id=3, name='小牛', hobby='篮球', color='黑色', age=33\}\]\}

### 2.对分组的条件（key）做定义将对象进行分组 ###

按年纪是否大于十八岁分组

Map<String, List<Student>> collect1 = studentList.stream().collect(groupingBy(key -> {
    
    if (key.getAge() > 18) {
    
    return "成年";
    
    } else {
    
    return "未成年";
    
    }
    
    }));

打印Map结果：

\{未成年=\[Student\{id=2, name='小白', hobby='足球', color='蓝色', age=11\}, Student\{id=4, name='小羊', hobby='足球', color='红色', age=17\}\],

成年=\[Student\{id=1, name='小红', hobby='篮球', color='红色', age=22\}, Student\{id=3, name='小牛', hobby='篮球', color='黑色', age=33\}, Student\{id=5, name='小绿', hobby='足球', color='蓝色', age=24\}\]\}

### 3.在分组时对数据进行过滤 ###

过滤掉不想要的数据时,

在java8，需在分组前对集合进行过滤，再分组。但会导致一类key中的元素都被过滤掉以后,分组结果不会有该key.

Map<String, List<Student>> collect = studentList.stream()
    .filter(f->!"红色".equals(f.getColor()))
    .collect(groupingBy(Student::getColor));

打印Map结果：

\{黑色=\[Student\{id=3, name='小牛', hobby='篮球', color='黑色', age=33\}\],

蓝色=\[Student\{id=2, name='小白', hobby='足球', color='蓝色', age=11\}, Student\{id=5, name='小绿', hobby='足球', color='蓝色', age=24\}\]\}

java9新增方法 filtering（）方法，可以在分组时进行过滤，以避免分组的结果map，因为过滤掉了一种key内的所有元素导致map中没有该key

filtering()

Map<String, List<Student>> collect = studentList.stream()
    .collect(groupingBy(Student::getColor,
    filtering(f->!"红色".equals(f.getColor()),toList())));

打印Map结果：

\{黑色=\[Student\{id=3, name='小牛', hobby='篮球', color='黑色', age=33\}\],

蓝色=\[Student\{id=2, name='小白', hobby='足球', color='蓝色', age=11\}, Student\{id=5, name='小绿', hobby='足球', color='蓝色', age=24\}\],

红色=\[\]\}

### 4.在分组时,将收集的对象映射处理为其他值 ###

Map<String, List<String >> collect = studentList.stream()
    .collect(groupingBy(Student::getColor,mapping(Student::getName,toList())));

打印Map结果：

\{黑色=\[小牛\],

蓝色=\[小白, 小绿\],

红色=\[小红, 小羊\]\}

或将mapping()替换为其他方法对子组的数据进行统计处理,比如maxBy()求子组中的Optional(最大值),或collectingAndThen(maxBy(),Optional::get)去除外层的Optional,直接获得最大值.

或者替换为求和,sumInt等

### 5.在分组时,如果收集的value为list中包list,想将两层list转为一层,可使用flatmapping()方法 ###

java9新增方法

和java8的flatmap()方法一样,通过将需要收集的集合元素A收集到总集合B时,将A中的单个集合转为Stream,和B的Stream通过flatmap()方法,将中间A的Stream去掉,当A中的单个集合中的元素都拿出来放到B中.

两层

Map<String,List<List<String>>> 将两层List拆开 Map<String,List<String>>

Map<String, List<List<String >>> collect = studentList.stream()
    .collect(groupingBy(Student::getHobby,
    mapping(m->colorToMindMap.get(m.getColor()),toList())));
    
    
    
    Map<String, List<String >> collect = studentList.stream()
    .collect(groupingBy(Student::getHobby,
    flatMapping(f->colorToMindMap.get(f.getColor()).stream(),toList())));

### 6.多级分组 ###

所谓多级分组,就是在groupingBy(a,b)的第二个参数继续放groupingBy(a,b)方法.规则和原来一样.

在平时用的groupingBy(Student::getColor)内部调用的其实就是groupingBy(Student::getColor,toList()), toList()方法就代表着将元素按list的形式来存储到对应的key下,改成groupBy(a,b)就是以分组的方式继续存储.