java 集合根据某多个字段进行分组

原创

mob64ca12ea10ec 2024-10-17 13:54:26 ©著作权

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

Java集合：根据多个字段分组

在Java开发中，处理集合数据时，常常需要对集合中的元素进行分组。借助Java 8引入的Stream API，我们可以更简洁、更高效地完成这一任务。本文将逐步指导你通过示例，学习如何根据多个字段对Java集合进行分组。

流程概览

以下是实现步骤的概览：

步骤	说明
1. 定义数据类	创建一个表示要分组的数据的类
2. 创建集合	创建一个包含数据的集合
3. 使用Stream	使用Java 8的Stream API对数据进行分组
4. 输出结果	将分组后的结果输出到控制台

每一步的实现

步骤 1: 定义数据类

首先，我们需要定义一个数据类，以便后续分组操作。下面是一个Person类，它包含姓名、年龄和城市三个字段：

// 定义Person类
public class Person {
    private String name; // 姓名
    private int age;     // 年龄
    private String city; // 城市

    // 构造函数
    public Person(String name, int age, String city) {
        this.name = name;
        this.age = age;
        this.city = city;
    }

    // getter方法
    public String getName() {
        return name;
    }

    public int getAge() {
        return age;
    }

    public String getCity() {
        return city;
    }
}

步骤 2: 创建集合

接下来，我们创建一个List<Person>集合，并向其中添加数据：

import java.util.ArrayList;
import java.util.List;

// 创建一个包含Person对象的列表
List<Person> people = new ArrayList<>();
people.add(new Person("Alice", 30, "New York"));
people.add(new Person("Bob", 25, "Los Angeles"));
people.add(new Person("Charlie", 30, "New York"));
people.add(new Person("David", 25, "Chicago"));

步骤 3: 使用Stream API进行分组

现在，我们将使用Stream API对这些数据进行分组。例如，我们可以根据年龄和城市来分组：

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

// 使用Stream进行分组
Map<String, List<Person>> groupedByCityAndAge = people.stream()
    .collect(Collectors.groupingBy(person -> person.getCity() + "-" + person.getAge()));

// 以上代码将根据城市和年龄进行分组，返回一个Map，key为"城市-年龄"

步骤 4: 输出结果

最后，输出分组后的结果：

// 输出分组结果
groupedByCityAndAge.forEach((key, value) -> {
    System.out.println("Group: " + key);
    value.forEach(person -> {
        System.out.println(" - " + person.getName());
    });
});

序列图

以下是整个流程的序列图：

sequenceDiagram
    participant Developer
    participant Code
    participant Console

    Developer->>Code: 定义Person类
    Developer->>Code: 创建包含数据的集合
    Developer->>Code: 使用Stream API分组数据
    Code->>Console: 输出分组结果