java老式的分组方式(对list for循环然后 if判断 放入map) 代码复杂,易读性差,维护性差,故本文汇总了Stream流中的分组方法供大家参考,如对您有帮助,请抬抬高贵的小手点个赞吧,欢迎大佬留下高见
(以下方法默认都是java8的方法,java9新增方法有标注)
List studentList = Arrays.asList(
new Student(1L,"小红","篮球","红色",22),
new Student(2L,"小白","足球","蓝色",11),
new Student(3L,"小牛","足球","黑色",33),
new Student(4L,"小羊","足球","红色",17),
new Student(5L,"小绿","足球","蓝色",24));
按颜色分组
Map> 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}]}
按年纪是否大于十八岁分组
Map> 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}]}
过滤掉不想要的数据时,
在java8,需在分组前对集合进行过滤,再分组。但会导致一类key中的元素都被过滤掉以后,分组结果不会有该key.
Map> 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> 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}],
红色=[]}
Map> collect = studentList.stream()
.collect(groupingBy(Student::getColor,mapping(Student::getName,toList())));
打印Map结果:
{黑色=[小牛],
蓝色=[小白, 小绿],
红色=[小红, 小羊]}
或将mapping()替换为其他方法对子组的数据进行统计处理,比如maxBy()求子组中的Optional(最大值),或collectingAndThen(maxBy(),Optional::get)去除外层的Optional,直接获得最大值.
或者替换为求和,sumInt等
java9新增方法
和java8的flatmap()方法一样,通过将需要收集的集合元素A收集到总集合B时,将A中的单个集合转为Stream,和B的Stream通过flatmap()方法,将中间A的Stream去掉,当A中的单个集合中的元素都拿出来放到B中.
两层
Map
Map>> collect = studentList.stream()
.collect(groupingBy(Student::getHobby,
mapping(m->colorToMindMap.get(m.getColor()),toList())));
Map> collect = studentList.stream()
.collect(groupingBy(Student::getHobby,
flatMapping(f->colorToMindMap.get(f.getColor()).stream(),toList())));
所谓多级分组,就是在groupingBy(a,b)的第二个参数继续放groupingBy(a,b)方法.规则和原来一样.
在平时用的groupingBy(Student::getColor)内部调用的其实就是groupingBy(Student::getColor,toList()), toList()方法就代表着将元素按list的形式来存储到对应的key下,改成groupBy(a,b)就是以分组的方式继续存储.