Collectors.groupingBy可用于分组统计,如按部门统计员工数量:Map countByDept = employees.stream().collect(Collectors.groupingBy(Employee::getDepartment, Collectors.counting()));
在Java中,可以使用 Collectors.groupingBy 对集合中的数据按指定条件进行分组,并统计每组的数量或汇总信息。这个方法通常与 Stream API 配合使用,适合处理 List、Set 等集合类型的数据。
1. 基本语法和用法
Collectors.groupingBy 有三种常见重载形式:
-
groupingBy(Function):根据函数返回值分组,结果是 Map>
-
groupingBy(Function, Collector):在分组后对每组数据应用另一个收集器,比如 counting() 统计数量
-
groupingBy(Function, Supplier, Collector):可指定返回 Map 的类型(如 LinkedHashMap)
2. 按字段分组并统计数量
假设有一个员工类:
class Employee { private String department; private String name; // 构造函数、getter等省略}
现在要按部门统计员工人数:
Map countByDept = employees.stream() .collect(Collectors.groupingBy(Employee::getDepartment, Collectors.counting()));
结果是一个 Map,key 是部门名,value 是该部门员工数量。
3. 多级分组(嵌套分组)
如果想先按部门分组,再在每个部门内按姓名长度分组:
Map>> result = employees.stream() .collect(Collectors.groupingBy( Employee::getDepartment, Collectors.groupingBy(emp -> emp.getName().length()) ));
这会生成一个两级 Map:外层 key 是部门,内层 key 是名字长度。
4. 分组后做其他聚合操作
除了统计数量,还可以求和、取平均值等:
- 统计每组人数:
Collectors.counting()
- 计算工资总和:
Collectors.summingDouble(Employee::getSalary)
- 获取每组员工名单:
Collectors.toList()
- 去重后名单:
Collectors.toSet()
例如,统计每个部门的平均工资:
Map avgSalaryByDept = employees.stream() .collect(Collectors.groupingBy( Employee::getDepartment, Collectors.averagingDouble(Employee::getSalary) ));
基本上就这些。灵活组合 groupingBy 和下游收集器,能应对大多数分组统计场景。关键是理解函数式接口如何提取分组依据,以及如何嵌套使用收集器。