java中stream流分组

Java中Stream流分组

引言

在Java中，Stream是一种用于操作集合元素的高级抽象。Stream提供了一种函数式编程的方式来处理集合数据，可以快速、简洁地操作集合数据。除了常见的map、filter、reduce等操作外，Stream还提供了分组的功能，即根据指定的条件将集合元素分组。

本文将介绍Java中Stream流的分组功能，包括如何使用Collectors.groupingBy方法进行分组，以及如何处理分组后的数据。

Stream流分组示例

假设我们有一个包含学生信息的集合，并且每个学生都有一个分数属性。现在我们希望根据学生的分数将其分成不同的等级，即将分数在60以下的学生分为不及格等级，分数在60到70之间的学生分为及格等级，分数在70到80之间的学生分为中等等级，分数在80以上的学生分为优秀等级。

首先，我们创建一个Student类表示学生信息：

public class Student {
    private String name;
    private int score;

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

    // 省略getter和setter方法
}

然后，我们创建一个包含学生信息的集合：

List<Student> students = new ArrayList<>();
students.add(new Student("John", 85));
students.add(new Student("Alice", 75));
students.add(new Student("Bob", 65));
students.add(new Student("Tom", 55));
students.add(new Student("Lucy", 95));

接下来，我们可以使用Collectors.groupingBy方法对学生进行分组：

Map<String, List<Student>> groupedStudents = students.stream()
        .collect(Collectors.groupingBy(student -> {
            if (student.getScore() < 60) {
                return "不及格";
            } else if (student.getScore() < 70) {
                return "及格";
            } else if (student.getScore() < 80) {
                return "中等";
            } else {
                return "优秀";
            }
        }));

在上述代码中，我们使用Collectors.groupingBy方法对学生进行分组。我们通过lambda表达式指定了分组的条件，即根据学生的分数返回对应的等级名称。

分组后，我们可以通过遍历Map来输出分组后的结果：

for (Map.Entry<String, List<Student>> entry : groupedStudents.entrySet()) {
    String grade = entry.getKey();
    List<Student> studentsInGrade = entry.getValue();
    System.out.println("等级：" + grade);
    for (Student student : studentsInGrade) {
        System.out.println("姓名：" + student.getName() + "，分数：" + student.getScore());
    }
}

上述代码中，我们使用entrySet方法遍历分组后的Map，并输出每个等级及对应的学生信息。

流程图

flowchart TD
    A[开始] --> B[创建学生集合]
    B --> C[分组]
    C --> D[遍历分组结果并输出]
    D --> E[结束]

结论

通过使用Java中的Stream流分组功能，我们可以方便地根据指定的条件对集合元素进行分组。在本文中，我们以学生分数为例，演示了如何使用Collectors.groupingBy方法对学生进行分组，并输出分组后的结果。希望本文对你理解Java中Stream流分组的使用有所帮助。