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

背景

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

Java 8 引入了 Stream API,它为处理集合数据提供了一个声明性的方法。Stream API 提供了一系列操作来对数据进行转换和聚合,其中 toMap 和 groupingBy 是两个常用的终端操作(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 不要求键唯一,因为每个键对应的是一个元素列表。

总结

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

相关推荐
weixin_39938069几秒前
TongWeb7.0.4.9M11如何接入到idea(by lqw)
java·ide·intellij-idea
paopaokaka_luck12 分钟前
社区服务平台(加权协同过滤推荐、Jieba中文词频分析、WebSocket实时聊天、求助处理闭环、 活动积分闭环、ECharts多维分析)
java·前端·spring boot·mysql·echarts·mybatis
.Bryce.19 分钟前
T2-LLM评测不等于Agent评测
android·java·人工智能·后端·gpt·agent·1024程序员节
Wang's Blog20 分钟前
Java 项目实战: 交友项目 - 用户体系集成:查询用户环信账号密码
java·交友
Wang's Blog32 分钟前
Java 项目实战: 交友项目 - 发布评论
java·开发语言·交友
qq_364682151 小时前
录音APP怎么切换语言?多语言版本如何使用
java·服务器·前端
Wang's Blog1 小时前
Java 项目实战: 交友项目 - 用户体系集成:根据环信id查询用户详情
java·lua·交友
摇滚侠1 小时前
《On Java 中文版 基础卷》阅读笔记 复用 13
java·笔记·python
2601_968900771 小时前
玻璃深加工的工艺窗口:4 道工序的温度与时间参数边界
android·java·javascript
薛之谦_1 小时前
用 Java 实现阿里云式 IAM 子用户管理:架构设计与表结构详解
java·python·阿里云