Java中按条件拆分集合推荐用Collectors.groupingBy()配合Lambda,支持单条件(如布尔值或语义化字符串)、多条件(record或字符串拼接)分组,可嵌套收集器取每组前N条,并需妥善处理null值。
Java中根据条件拆分集合,核心是按某种规则把元素划分到不同子集合里,本质就是分组。JDK 8+ 推荐用 Collectors.groupingBy() 配合 Lambda 实现,简洁、安全、可读性强。
最常见场景:比如把用户列表按年龄区间、状态、类型等字段归类。
示例:将 List
Map> grouped = users.stream() .collect(Collectors.groupingBy(user -> user.getAge() >= 18));
结果得到一个 Map,key 是 true/false,value 是对应条件成立的用户列表。
user -> user.getAge() >= 18 ? "adult" : "minor"
当需要“城市 + 性别”或“部门 + 职级”这类复合维度时,可以构造一个临时 key(推荐用 record 或简单字符串拼接)。
示例:按城市和会员等级分组(使用 record 避免手写 equals/hashCode):
record LocationLevel(String city, String level) {}
Map> grouped = users.stream()
.collect(Collectors.groupingBy(u -> new LocationLevel(u.getCity(), u.getLevel())));
或者用字符串拼接(适合简单场景,注意避免分隔符冲突):
.collect(Collectors.groupingBy(u -> u.getCity() + "|" + u.getLevel()))
groupingBy 默认返回完整子列表,如果只需每组头几条,可嵌套收集器:
Map> top3ByCity = users.stream() .collect(Collectors.groupingBy( User::getCity, Collectors.collectingAndThen( Collectors.toList(), list -> list.stream().limit(3).collect(Collectors.toList()) ) ));
也可以用 Collectors.takeWhile 或自定义 Collector,但 limit + collectingAndThen 更直观。
原始数据含 null 字段?groupingBy 默认抛 NullPointerException。两种稳妥做法:
.filter(Objects::nonNull).filter(u -> u.getCity() != null)
u -> Optional.ofNullable(u.getCity()).orElse("unknown")
如果某组可能为空(比如按枚举值分组但某些值没数据),可用 Collectors.groupingBy(..., HashMap::new, ...) 指定 map 工厂,再结合 computeIfAbsent 补默认空列表(视业务而定)。
基本上就这些。不复杂但容易忽略空值和 key 设计——选对 key 类型,分组逻辑自然清晰。