java stream怎么实现分组

原创

mob64ca12dea1dc 2023-12-11 08:44:55 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12dea1dc的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java Stream如何实现分组

在Java中，Stream是一种函数式编程的概念，它提供了一种便捷的方式来处理集合数据。Stream可以用于各种集合操作，包括过滤、映射、排序等。其中一个常见的需求是对集合数据进行分组操作。本文将介绍如何使用Java Stream来实现分组，并给出具体的代码示例。

需求描述

假设我们有一个包含学生信息的集合，每个学生有姓名和年龄两个属性。我们希望将学生按照年龄进行分组，即将年龄相同的学生放在同一个组中。

解决方案

Java Stream提供了一个Collectors.groupingBy方法，可以方便地对集合进行分组操作。我们可以使用这个方法来实现对学生集合的分组。

首先，我们需要定义一个Student类，包含姓名和年龄两个属性。代码如下所示：

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

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

    public String getName() {
        return name;
    }

    public int getAge() {
        return age;
    }
}

接下来，我们创建一个包含学生信息的集合，并使用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", 18),
                new Student("Bob", 20),
                new Student("Charlie", 18),
                new Student("David", 20),
                new Student("Eva", 18)
        );

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

        System.out.println(groupedStudents);
    }
}

在上述代码中，我们使用Collectors.groupingBy方法将学生按照年龄进行分组。Student::getAge表示使用学生的年龄作为分组的依据。groupedStudents是一个Map<Integer, List<Student>>类型的变量，其中Key为年龄，Value为对应年龄的学生列表。

运行上述代码，输出结果如下所示：

{
    18=[Student{name='Alice', age=18}, Student{name='Charlie', age=18}, Student{name='Eva', age=18}], 
    20=[Student{name='Bob', age=20}, Student{name='David', age=20}]
}

可以看到，学生已经按照年龄成功分组。

类图

以下是Student类的类图：

classDiagram
    class Student {
        -String name
        -int age
        +Student(name: String, age: int)
        +getName(): String
        +getAge(): int
    }