有时我们需要对集合进行分组操作,这时可以使用Java8提供的Stream方式进行分组。挺好用的,此处记录下。直接贴code:
Road实体:
@Data
@NoArgsConstructor
@AllArgsConstructor
public class Road {/**
* 名称
*/
private String name;/**
* 道路长度
*/
private double len;/**
* 社区划分
*/
private String community;
}
Stream分组:
Road road1 = new Road("荷花大道_东段", 100.2, "荷花社区");
Road road2 = new Road("荷花大道西段", 88.2, "荷花社区");
Road road3 = new Road("剑北大道_东段", 99.2, "剑北社区");
Road road4 = new Road("剑北大道西段", 111.4, "剑北社区");
Road road5 = new Road("剑北大道_南段", 68.9, "剑北社区")
List<Road> roadList = new ArrayList<>();
roadList.add(road1);
roadList.add(road2);
roadList.add(road3);
roadList.add(road4);
roadList.add(road5);
// ①根据 社区属性 分组
// 两种方式皆可
Map<String, List<Road>> roadMap1 = roadList.stream().collect(Collectors.groupingBy(Road::getCommunity));
Map<String, List<Road>> roadMap2 = roadList.stream().collect(Collectors.groupingBy(Road::getCommunity, Collectors.toList()));
// ②将 道路长度大于100 的分为一组
Map<String, List<Road>> roadMap3 = roadList.stream().collect(Collectors.groupingBy(x -> {
if (x.getLen() > 100) {
return "大于100";
} else {
return "小于等于100";
}
}));
// ③想得到一个通过道路长度分组并有序的结果(此处道路长度都不一样,没必要分组,只是想体现有序的返回)
TreeMap<Double, List<Road>> roadMap4 = roadList.stream().collect(Collectors.groupingBy(Road::getLen, TreeMap::new, Collectors.toList()));
// ④将名字中的 大道 替换为'' 并进行分组(主要是这个,平时用的时候,有时会搞忘)
Map<String, List<Road>> roadMap5 = roadList.stream().collect(Collectors.groupingBy(x -> x.getName().replace("_", ""), Collectors.toList()));
// 按自己习惯打印输出,roadMap1...roadMap5,进行结果验证比对
代码直接复制即可运行,可进行结果验证比对。
既然选择了远方,便只顾风雨兼程。-- 烟沙九洲