Java里如何使用Stream.flatMap扁平化集合_Stream扁平化技巧说明

P粉602998670 2025-11-26 00:00:00 次阅读

flatMap用于将嵌套集合展平为单一流，如将List转为List，或从用户列表中提取所有邮箱并过滤去重，其核心是将每个元素映射为St
ream后再合并，需注意空集合判空处理，避免NullPointerException。

在Java中，Stream.flatMap 是处理嵌套集合时非常实用的工具。它能将多个子集合“拍平”成一个单一的流，便于后续统一处理。比如你有一组用户，每个用户有多个邮箱，你想提取出所有邮箱并去重或过滤，这时 flatMap 就派上用场了。

flatMap 的基本作用

flatMap 和 map 不同，map 是一对一转换，而 flatMap 是一对多，并自动将多个结果合并成一个流。它的核心签名是：

Stream flatMap(Function super T, ? extends Stream extends R>> mapper)

也就是说，它接收一个函数，该函数把每个元素转为一个 Stream，然后把这些 Stream 内部的元素全部“摊开”合并到一个新流中。

常见使用场景：扁平化二维集合

假设有一个 List>，想把它变成一个包含所有字符串的一维列表：

List> listOfLists = Arrays.asList(
    Arrays.asList("a", "b"),
    Arrays.asList("c", "d"),
    Arrays.asList("e")
);

List flattened = listOfLists.stream()
    .flatMap(List::stream)
    .collect(Collectors.toList());

// 结果: ["a", "b", "c", "d", "e"]

这里 .flatMap(List::stream) 把每一个子 List 转成一个 Stream，然后拼接成整体流。

实际业务示例：用户与邮箱扁平化

考虑一个更贴近实际的例子：每个用户有多个联系方式，我们想提取所有邮箱地址：

class User {
    private String name;
    private List emails;

    // 构造方法、getter等省略
}

List users = ... ; // 假设已初始化

List allEmails = users.stream()
    .flatMap(user -> user.getEmails().stream())
    .distinct()
    .filter(email -> email.endsWith("@example.com"))
    .collect(Collectors.toList());

这个例子展示了 flatMap 的强大之处：从对象集合中提取内部集合，并进行去重、过滤等操作。