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 则适用于需要对数据进行分类和分组的情况。理解它们的使用方法和区别,有助于更高效地处理集合数据。

相关推荐
2401_85439108几秒前
Spring Boot OA:企业数字化转型的利器
java·spring boot·后端
山山而川粤8 分钟前
废品买卖回收管理系统|Java|SSM|Vue| 前后端分离
java·开发语言·后端·学习·mysql
栗豆包10 分钟前
w053基于web的宠物咖啡馆平台的设计与实现
java·struts·spring·tomcat·maven·intellij-idea
weixin_4467077439 分钟前
IDEA2024 maven构建跳过测试
java·maven
开朗觉觉1 小时前
RabbitMQ高可用&&延迟消息&&惰性队列
java·rabbitmq·java-rabbitmq
zmd-zk1 小时前
flink学习(3)——方法的使用—对流的处理(map,flatMap,filter)
java·大数据·开发语言·学习·flink·tensorflow
昵称20211 小时前
flink1.16+连接Elasticsearch7官方例子报错解决方案
java·flink·es7
爱编程的小生1 小时前
Easyexcel(6-单元格合并)
java·excel
小白不太白9501 小时前
设计模式之 迭代器模式
java·设计模式·迭代器模式
闲人一枚(学习中)1 小时前
设计模式-创建型-单例模式
java·单例模式·设计模式