Java8的stream之groupingBy()分组排序
- groupingBy()是Stream API中最强大的收集器Collector之一,提供与SQL的GROUP BY子句类似的功能。
- 需要指定一个属性才能使用,通过该属性执行分组。我们通过提供功能接口的实现来实现这一点 - 通常通过传递lambda表达式。
一、默认升序排序
stationMap = peopleList.stream().collect(Collectors.groupingBy(People::getJgId));
1.descendingMap()降序排序
stationMap = peopleList.stream().collect(Collectors.groupingBy(People::getJgId,TreeMap::new,Collectors.toList())).descendingMap();
2.总结
TreeMap默认按照key升序排序,collectPlan.descendingMap()可以进行降序排序
List<People> peopleList = Lists.newArrayList();
peopleList.add(new People(1, "小王", 5));
peopleList.add(new People(1, "小李", 4));
peopleList.add(new People(2, "小张", 3));
peopleList.add(new People(2, "小皇", 2));
peopleList.add(new People(2, "小刘", 1));
//分组排序并加序号
Map<Integer, List<People>> stationMap = new HashMap<Integer, List<People>>();
// 使用stream流将list转为map,key为指定字段的值
stationMap = peopleList.stream().collect(Collectors.groupingBy(People::getJgId));
log.info(stationMap.toString());
stationMap = peopleList.stream().collect(Collectors.groupingBy(People::getJgId,TreeMap::new,Collectors.toList())).descendingMap();
log.info(stationMap.toString());
Stream
中的 groupingBy
是一个非常有用的收集器,它允许我们根据给定的分类函数将流中的元素分组。除了基本的用法外,groupingBy
还可以与其他收集器组合,以实现更复杂的分组和聚合操作。下面是一些 groupingBy
的多种用法示例:
二、基本用法
最基本的用法是根据对象的某个属性进行分组。
假设有一个 Person
类:
class Person {
private String name;
private int age;
// 省略构造函数和getter/setter
}
现在我们想根据 age
属性来分组 Person
对象的列表:
List<Person> persons = Arrays.asList(
new Person("Alice", 23),
new Person("Bob", 25),
new Person("Charlie", 23)
);
Map<Integer, List<Person>> groupedByAge = persons.stream()
.collect(Collectors.groupingBy(Person::getAge));
System.out.println(groupedByAge);
输出结果将是:
{23=[Person{name='Alice', age=23}, Person{name='Charlie', age=23}],
25=[Person{name='Bob', age=25}]}
这里 groupingBy(Person::getAge)
将 Person
对象按照他们的 age
属性进行了分组。
1. 结合 mapping 收集器
有时我们想要对分组后的结果进行进一步的映射和处理。比如,将分组后的每组 Person
对象转换为他们的名字列表:
Map<Integer, List<String>> nameGroupsByAge = persons.stream()
.collect(Collectors.groupingBy(
Person::getAge,
Collectors.mapping(Person::getName, Collectors.toList())
));
System.out.println(nameGroupsByAge);
这里的 mapping(Person::getName, Collectors.toList())
将每个 Person
对象映射为他们的名字,并将结果收集到一个列表中。
2. 使用 downstream 收集器
有时需要在分组后对每个组进行更复杂的聚合操作,比如计算每个年龄组的人数或者其他统计量。可以使用 downstream
收集器来实现这一点:
Map<Integer, Long> countByAge = persons.stream()
.collect(Collectors.groupingBy(
Person::getAge,
Collectors.counting()
));
System.out.println(countByAge);
这里的 Collectors.counting()
是一个用于计数的内置收集器,它统计了每个年龄组的元素数量。
3. 自定义分组函数
有时候可能需要更复杂的逻辑来决定如何分组。可以使用 groupingBy
的第二个参数来传递一个自定义的 Collector
或者 supplier
函数来处理分组逻辑。例如,将人按照年龄分成青少年、成年人和老年人:
Map<String, List<Person>> customGroups = persons.stream()
.collect(Collectors.groupingBy(
person -> {
if (person.getAge() < 18) {
return "青少年";
} else if (person.getAge() < 65) {
return "成年人";
} else {
return "老年人";
}
}
));
System.out.println(customGroups);
这里根据 person
的年龄将他们分成三组,并使用了一个自定义的分类函数。
4.总结
groupingBy
收集器提供了强大的功能,可以根据指定的分类函数对流中的元素进行分组。结合其他收集器,如 mapping
和 downstream
,可以实现丰富的数据处理操作,非常适合在需要对数据进行分组和聚合的场景中使用。