java8 stream 多字段 group-CFANZ编程社区

Java8 Stream 多字段 Group

在Java 8中引入的Stream API为我们提供了一种强大而简洁的方式来处理集合数据。其中一个常见的需求是按照多个字段对数据进行分组。在本文中，我们将探讨如何使用Java 8 Stream来实现多字段分组，并给出相应的代码示例。

什么是多字段分组

在数据处理中，我们经常需要根据多个字段对数据进行分类或分组。例如，假设我们有一个包含学生信息的列表，每个学生有姓名、班级和年龄等字段。我们可能希望按照班级和年龄对学生进行分组，以便更好地分析数据。这就是多字段分组的概念。

使用Java 8 Stream进行多字段分组

Java 8中的Stream API提供了一个groupingBy方法，可以对数据进行分组。对于多字段分组，我们可以通过传递一个包含多个字段的Collector来实现。

让我们以一个简单的示例开始。假设我们有一个Student类，包含姓名、班级和年龄字段。

public class Student {
    private String name;
    private String className;
    private int age;

    // 构造方法、getter和setter省略
}

我们创建一个包含多个学生对象的列表：

List<Student> students = Arrays.asList(
    new Student("Alice", "Class A", 20),
    new Student("Bob", "Class B", 21),
    new Student("Alice", "Class A", 19),
    new Student("Bob", "Class A", 20),
    new Student("Charlie", "Class B", 19)
);

现在，我们希望按照班级和年龄对学生进行分组。我们可以使用以下代码来实现：

Map<String, Map<Integer, List<Student>>> groupedStudents = students.stream()
    .collect(Collectors.groupingBy(Student::getClassName,
             Collectors.groupingBy(Student::getAge)));

在上述代码中，我们首先调用stream方法将列表转换为流。然后，我们使用groupingBy方法按照班级对学生进行第一层分组，并使用groupingBy方法按照年龄对学生进行第二层分组。最终的结果是一个Map<String, Map<Integer, List<Student>>>，其中第一层的键是班级，第二层的键是年龄，值是学生列表。

示例代码

下面是一个完整的示例代码，演示如何使用Java 8 Stream进行多字段分组：

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class Main {
    public static void main(String[] args) {
        List<Student> students = Arrays.asList(
            new Student("Alice", "Class A", 20),
            new Student("Bob", "Class B", 21),
            new Student("Alice", "Class A", 19),
            new Student("Bob", "Class A", 20),
            new Student("Charlie", "Class B", 19)
        );

        Map<String, Map<Integer, List<Student>>> groupedStudents = students.stream()
            .collect(Collectors.groupingBy(Student::getClassName,
                     Collectors.groupingBy(Student::getAge)));

        System.out.println(groupedStudents);
    }
}

class Student {
    private String name;
    private String className;
    private int age;

    public Student(String name, String className, int age) {
        this.name = name;
        this.className = className;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public String getClassName() {
        return className;
    }

    public int getAge() {
        return age;
    }
}

以上代码将输出以下结果：

{
    Class A={
        19=[Student@1f32e575],
        20=[Student@3669bf35, Student@3eac0a99]
    },
    Class B={
        19=[Student@5a07e868],
        21=[Student@7ef20235]
    }
}

我们可以看到学生对象按照班级和年龄被正确地进行了分组。