项目中已经通过处理逻辑获得了一个列表,后续需要根据列表的一个属性或多个属性对其分组,可以采用 Java8 的 Stream 实现。
举例说明如下,创建一个 Java Class,名为 Student,其有四个属性,分别是:学号(stuNo)、姓名(stuNm)、年龄(age)、性别(sex)。现有一个学生列表,分别根据年龄(单一属性)、年龄和性别(多属性)进行分组。
Student 类定义如下:
package com.test.model;
import lombok.Getter;
@Getter
public class Student {
// 学号
private String stuNo;
// 姓名
private String stuNm;
// 年龄
private String age;
// 性别
private String sex;
public Student(String stuNo, String stuNm, String age, String sex) {
this.stuNo = stuNo;
this.stuNm = stuNm;
this.age = age;
this.sex = sex;
}
}
构造集合列表,用于后续测试:
package com.test.model;
import java.util.ArrayList;
import java.util.List;
public class ListStu {
// 构造测试数据,返回学生列表
public static List<Student> listStu() {
List<Student> stuList = new ArrayList<>();
stuList.add(new Student("2001", "段誉", "20", "男"));
stuList.add(new Student("2002", "乔峰", "20", "男"));
stuList.add(new Student("2003", "虚竹", "22", "男"));
stuList.add(new Student("2004", "张无忌", "22", "男"));
stuList.add(new Student("2005", "阿朱", "22", "女"));
stuList.add(new Student("2006", "王语嫣", "22", "女"));
return stuList;
}
}
一、根据单一属性(年龄:age)对学生列表进行分组并简单排序
package com.test.model;
import com.alibaba.fastjson.JSON;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class Test {
public static void main(String[] args) {
List<Student> stuList = ListStu.listStu();
// ------------- 将 stuList 按照 Student 的 age 属性进行分组,转为 map ------
Map<String, List<Student>> groupedMap =
stuList.stream().collect(Collectors.groupingBy(Student::getAge));
System.out.println("按年龄分组后的数据:");
System.out.println(JSON.toJSONString(groupedMap));
System.out.println("年龄为20的学生:");
System.out.println(JSON.toJSONString(groupedMap.get("20")));
// --------------- 将 stuList 按照 Student 的 age 属性进行分组,并按照 age 升序排序 --------------
TreeMap<String, List<Student>> treeMap = stuList.stream()
.collect(Collectors.groupingBy(Student::getAge, TreeMap::new, Collectors.toList()));
System.out.println("分组并按年龄升序排序后的 treeMap:");
System.out.println(JSON.toJSONString(treeMap));
}
}
执行结果如下:
按年龄分组后的数据:
{“22”:[{“age”:“22”,“sex”:“男”,“stuNm”:“虚竹”,“stuNo”:“2003”},{“age”:“22”,“sex”:“男”,“stuNm”:“张无忌”,“stuNo”:“2004”},{“age”:“22”,“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}],“20”:[{“age”:“20”,“sex”:“男”,“stuNm”:“段誉”,“stuNo”:“2001”},{“age”:“20”,“sex”:“男”,“stuNm”:“乔峰”,“stuNo”:“2002”}]}
年龄为20的学生:
[{“age”:“20”,“sex”:“男”,“stuNm”:“段誉”,“stuNo”:“2001”},{“age”:“20”,“sex”:“男”,“stuNm”:“乔峰”,“stuNo”:“2002”}]
分组并按年龄升序排序后的 treeMap:
{“20”:[{“age”:“20”,“sex”:“男”,“stuNm”:“段誉”,“stuNo”:“2001”},{“age”:“20”,“sex”:“男”,“stuNm”:“乔峰”,“stuNo”:“2002”}],“22”:[{“age”:“22”,“sex”:“男”,“stuNm”:“虚竹”,“stuNo”:“2003”},{“age”:“22”,“sex”:“男”,“stuNm”:“张无忌”,“stuNo”:“2004”},{“age”:“22”,“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}]}
二、根据多个属性(年龄和性别)对学生列表进行分组
方法1:嵌套调用 Java8 Collectors groupingBy 方法
package com.test.model;
import com.alibaba.fastjson.JSON;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class Test {
public static void main(String[] args) {
List<Student> stuList = ListStu.listStu();
// ------------------- 将 stuList 按照 Student 的 age 和 sex 属性进行分组 -------------------------
// 方法1:嵌套调用 Java8 Collectors groupingBy 方法
Map<String, Map<String, List<Student>>> map1 = stuList.stream()
.collect(Collectors.groupingBy(Student::getAge, Collectors.groupingBy(Student::getSex)));
System.out.println("按照年龄和性别分组后的数据:");
System.out.println(JSON.toJSONString(map1));
System.out.println("22岁的女学生:");
List<Student> list1 = map1.get("22").get("女");
System.out.println(JSON.toJSONString(list1));
}
}
执行结果如下:
按照年龄和性别分组后的数据:
{“22”:{“女”:[{“age”:“22”,“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}],“男”:[{“age”:“22”,“sex”:“男”,“stuNm”:“虚竹”,“stuNo”:“2003”},{“age”:“22”,“sex”:“男”,“stuNm”:“张无忌”,“stuNo”:“2004”}]},“20”:{“男”:[{“age”:“20”,“sex”:“男”,“stuNm”:“段誉”,“stuNo”:“2001”},{“age”:“20”,“sex”:“男”,“stuNm”:“乔峰”,“stuNo”:“2002”}]}}
22岁的女学生:
[{“age”:“22”,“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}]
方法2:多个属性拼接成一个组合属性
package com.test.model;
import com.alibaba.fastjson.JSON;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class Test {
public static void main(String[] args) {
List<Student> stuList = ListStu.listStu();
// ------------------- 将 stuList 按照 Student 的 age 和 sex 属性进行分组 -------------------------
// 方法2:多个属性拼接成一个组合属性
Map<String, List<Student>> map2 = stuList.stream().collect(Collectors.groupingBy(d -> fetchGroupKey(d)));
System.out.println("按照年龄和性别分组后的数据:");
System.out.println(JSON.toJSONString(map2));
System.out.println("22岁的女学生:");
System.out.println(JSON.toJSONString(map2.get("22_女")));
}
private static String fetchGroupKey(Student student) {
return student.getAge() + "_" + student.getSex();
}
}
执行结果如下:
按照年龄和性别分组后的数据:
{“20_男”:[{“age”:“20”,“sex”:“男”,“stuNm”:“段誉”,“stuNo”:“2001”},{“age”:“20”,“sex”:“男”,“stuNm”:“乔峰”,“stuNo”:“2002”}],“22_女”:[{“age”:“22”,“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}],“22_男”:[{“age”:“22”,“sex”:“男”,“stuNm”:“虚竹”,“stuNo”:“2003”},{“age”:“22”,“sex”:“男”,“stuNm”:“张无忌”,“stuNo”:“2004”}]}
22岁的女学生:
[{“age”:“22”,“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}]
方法3:在 Student 类中构造静态内部类,静态内部类的成员变量即为分组对应的多个属性
构造好静态内部类的 Student 类如下:
package com.test.model;
import lombok.Getter;
@Getter
public class Student {
// 学号
private String stuNo;
// 姓名
private String stuNm;
// 年龄
private String age;
// 性别
private String sex;
public Student(String stuNo, String stuNm, String age, String sex) {
this.stuNo = stuNo;
this.stuNm = stuNm;
this.age = age;
this.sex = sex;
}
// 静态内部类,含 age 和 sex 属性
public static class AgeSex {
public String age;
public String sex;
public AgeSex(String age, String sex) {
this.age = age;
this.sex = sex;
}
// 注意:因为 ageSex 对象会作为 Map 的 key,所以需要重写 equals 和 hashCode 方法,
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (!(o instanceof AgeSex)) return false;
AgeSex ageSex = (AgeSex) o;
if (age != null ? !age.equals(ageSex.age) : ageSex.age != null) return false;
return sex != null ? sex.equals(ageSex.sex) : ageSex.sex == null;
}
@Override
public int hashCode() {
int result = age != null ? age.hashCode() : 0;
result = 31 * result + (sex != null ? sex.hashCode() : 0);
return result;
}
}
public AgeSex getAgeSex() {
return new AgeSex(age, sex);
}
}
或者用 Lombok 的 @Data 注解为 Java 类自动生成 equals() 和 hashCode() 方法,示例如下:
package com.test.model;
import lombok.Data;
@Data
public class Student {
// 学号
private String stuNo;
// 姓名
private String stuNm;
// 年龄
private String age;
// 性别
private String sex;
public Student(String stuNo, String stuNm, String age, String sex) {
this.stuNo = stuNo;
this.stuNm = stuNm;
this.age = age;
this.sex = sex;
}
// 静态内部类,含 age 和 sex 属性
@Data
public static class AgeSex {
public String age;
public String sex;
public AgeSex(String age, String sex) {
this.age = age;
this.sex = sex;
}
}
public AgeSex getAgeSex() {
return new AgeSex(age, sex);
}
}
分组处理方法如下:
package com.test.model;
import com.alibaba.fastjson.JSON;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class Test {
public static void main(String[] args) {
List<Student> stuList = ListStu.listStu();
// ------------------- 将 stuList 按照 Student 的 age 和 sex 属性进行分组 ---------------------
// 方法3:在 Student 类里构造静态内部类(成员变量即分组对应的多个属性:age 和 sex)
Map<Student.AgeSex, List<Student>> map3 = stuList.stream()
.collect(Collectors.groupingBy(Student::getAgeSex));
System.out.println("按照年龄和性别分组后的数据:");
System.out.println(JSON.toJSONString(map3));
System.out.println("22岁的女学生:");
List<Student> list3 = map3.get(new Student.AgeSex("22", "女"));
System.out.println(JSON.toJSONString(list3));
}
}
执行结果如下所示:
按照年龄和性别分组后的数据:
{{“age”:“22”,“sex”:“男”}:[{“age”:“22”,“ageSex”:{“age”:“22”,“sex”:“男”},“sex”:“男”,“stuNm”:“虚竹”,“stuNo”:“2003”},{“age”:“22”,“ageSex”:{“age”:“22”,“sex”:“男”},“sex”:“男”,“stuNm”:“张无忌”,“stuNo”:“2004”}],{“age”:“20”,“sex”:“男”}:[{“age”:“20”,“ageSex”:{“age”:“20”,“sex”:“男”},“sex”:“男”,“stuNm”:“段誉”,“stuNo”:“2001”},{“age”:“20”,“ageSex”:{“age”:“20”,“sex”:“男”},“sex”:“男”,“stuNm”:“乔峰”,“stuNo”:“2002”}],{“age”:“22”,“sex”:“女”}:[{“age”:“22”,“ageSex”:{“age”:“22”,“sex”:“女”},“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“ageSex”:{“age”:“22”,“sex”:“女”},“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}]}
22岁的女学生:
[{“age”:“22”,“ageSex”:{“age”:“22”,“sex”:“女”},“sex”:“女”,“stuNm”:“阿朱”,“stuNo”:“2005”},{“age”:“22”,“ageSex”:{“age”:“22”,“sex”:“女”},“sex”:“女”,“stuNm”:“王语嫣”,“stuNo”:“2006”}]
以上是用 Java8 Stream 对集合进行分组的几种方法,参考了以下链接:
https://stackoverflow.com/questions/28342814/group-by-multiple-field-names-in-java-8#