Java8 - flatMap() 介绍

_板栗_2025-01-25 8:56

1.flatMap() 的作用与用法

在 Java8 中引入了 SteamAPI，其中有一个功能是 flatMap()。它的作用是将一个二维数组处理成一维数组，或者说是将一个 2 级 Stream 处理成 1 级 Stream。

举例来说，下面的二维数组可以通过 flatMap() 处理成一维数组。

java 复制代码

// 二维数据
Integer[][] twoDimArr = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
// 一维数据
Integer[] oneDimArr = {1,2,3,4,5,6,7,8,9}

处理过程为：

java 复制代码

Integer[][] twoDimArr = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
Integer[] onewDimArr = Arrays.stream(twoDimArr).flatMap(Arrays::stream).toArray(Integer[]::new);

这一过程也可以理解为将 2 级的流 Stream<Integer[]>展平成了 1 级的流 Stream<Integer>。

从语法上来看，我们在 flatMap 中要写的是一个 lambda 表达式（或上图中的方法引用），而这个表达式要做的是将一维数组（比如 Integer[]）转成流形式。

更多的例子可以参考 Java8 - Streams flatMap()

2.参考

Java8 - Streams flatMap()