Java Streams 使用 toMap 和 groupingBy 的方法及其异同

背景

今天在处理一个数据分组问题是,突然想到是使用 toMap 呢还是使用 groupBy 呢,因为这两个 API 都是比较常用的,一时语塞,后来想想还是比较好区分的,下面就介绍下这两个常用的 API。

Java 8 引入了 Stream API,它为处理集合数据提供了一个声明性的方法。Stream API 提供了一系列操作来对数据进行转换和聚合,其中 toMapgroupingBy 是两个常用的终端操作(terminal operation)。

1. toMap 操作

toMap 是 Stream 的一个收集器(collector),它将流中的元素转换为一个 Map。该方法常用于需要将流中的数据转换为键值对的场景。

用法示例:

java 复制代码
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class ToMapExample {
    public static void main(String[] args) {
        List<Student> students = Arrays.asList(
            new Student(1, "Alice"),
            new Student(2, "Bob"),
            new Student(3, "Charlie")
        );

        Map<Integer, String> studentMap = students.stream()
            .collect(Collectors.toMap(Student::getId, Student::getName));

        System.out.println(studentMap);
    }
}

@Data
class Student {
    private int id;
    private String name;

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

在这个示例中,我们将学生列表转换为一个以学生 ID 为键,学生名字为值的 Map

注意事项:

  • toMap 要求键是唯一的,如果流中存在重复键,会抛出 IllegalStateException

  • 可以传递第三个参数来解决键冲突:

java 复制代码
.collect(Collectors.toMap(Student::getId, Student::getName, (name1, name2) -> name1));

(name1, name2) -> name1) 表示键冲突时,保留前一个数据,类似的,可以保留后面的数据。

2. groupingBy 操作

groupingBy 是另一个收集器,它基于某个分类函数对流中的元素进行分组,返回一个 Map,键为分类依据,值为该分类下的元素列表。

用法示例:

java 复制代码
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class GroupingByExample {
    public static void main(String[] args) {
        List<Student> students = Arrays.asList(
            new Student(1, "Alice", "A"),
            new Student(2, "Bob", "B"),
            new Student(3, "Charlie", "A")
        );

        Map<String, List<Student>> studentGroups = students.stream()
            .collect(Collectors.groupingBy(Student::getGrade));

        System.out.println(studentGroups);
    }
}

@Data
class Student {
    private int id;
    private String name;
    private String grade;

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

在这个示例中,我们根据学生的年级对学生进行分组,生成一个以年级为键,学生列表为值的 Map

注意事项:

  • groupingBy 的键可以是任何类型,且键不需要唯一。
  • 可以通过 Collectors.mapping 等进一步进行嵌套收集。

异同

相同点:

  • 都是 Collectors 提供的静态方法。
  • 都是终端操作,用于将流中的数据收集到 Map 中。

不同点:

  • 目的不同toMap 用于将流中的数据转换为键值对,而 groupingBy 用于根据某个分类标准对数据进行分组。
  • 返回结果toMap 返回的是一个单层的 Map,键为唯一值;groupingBy 返回的是一个嵌套的 Map,键为分类标准,值为该分类下的元素列表。
  • 键的唯一性toMap 要求键唯一,若不唯一则需要处理冲突;groupingBy 不要求键唯一,因为每个键对应的是一个元素列表。

总结

返回的是一个嵌套的Map`,键为分类标准,值为该分类下的元素列表。

  • 键的唯一性toMap 要求键唯一,若不唯一则需要处理冲突;groupingBy 不要求键唯一,因为每个键对应的是一个元素列表。

总结

toMapgroupingBy 是 Java Stream API 中强大的收集器,它们在数据转换和分组处理中有着广泛的应用。toMap 适用于需要将流中的数据映射为键值对的情况,而 groupingBy 则适用于需要对数据进行分类和分组的情况。理解它们的使用方法和区别,有助于更高效地处理集合数据。

相关推荐
大模型玩家七七1 小时前
基于语义切分 vs 基于结构切分的实际差异
java·开发语言·数据库·安全·batch
寻星探路6 小时前
【深度长文】万字攻克网络原理:从 HTTP 报文解构到 HTTPS 终极加密逻辑
java·开发语言·网络·python·http·ai·https
曹牧8 小时前
Spring Boot:如何测试Java Controller中的POST请求?
java·开发语言
爬山算法9 小时前
Hibernate(90)如何在故障注入测试中使用Hibernate?
java·后端·hibernate
kfyty7259 小时前
集成 spring-ai 2.x 实践中遇到的一些问题及解决方案
java·人工智能·spring-ai
猫头虎9 小时前
如何排查并解决项目启动时报错Error encountered while processing: java.io.IOException: closed 的问题
java·开发语言·jvm·spring boot·python·开源·maven
李少兄9 小时前
在 IntelliJ IDEA 中修改 Git 远程仓库地址
java·git·intellij-idea
忆~遂愿9 小时前
ops-cv 算子库深度解析:面向视觉任务的硬件优化与数据布局(NCHW/NHWC)策略
java·大数据·linux·人工智能
小韩学长yyds9 小时前
Java序列化避坑指南:明确这4种场景,再也不盲目实现Serializable
java·序列化