当前位置：首页 > news >正文

java中flatMap用法

news 2025/6/28 3:55:16

java中map是把集合每个元素重新映射，元素个数不变，但是元素值发生了变化。而flatMap从字面上来说是压平这个映射，实际作用就是将每个元素进行一个一对多的拆分，细分成更小的单元，返回一个新的Stream流，新的流元素个数增加。

java官方给出的示例如下：

# 按行读取文件内容
Stream<String> lines = Files.lines(path);
# 将每一行文字按照空格拆分为单词
Stream<String> words = lines.flatMap(line -> Stream.of(line.split(" +")));

如下所示的文本：

hello world
this is a test txt
done

经过这个转换，得到的结果如下：

[hello, world, this, is, a, test, txt, done]

我们大概知道，flatMap压平的作用，就是把原来的三行字符串文字变为了一个流，流数组中每个元素是单词。

============

我们假设有这样的需求，有一个[[1,3],[2,4]]的二维数组，我们要得到一个一维数组[1,3,2,4]。那么这里的flatMap就正好排上用场，原来是二维的，现在压平为一维，正好分解了。

List<Integer> lista = new ArrayList<>();
lista.add(1);
lista.add(3);List<Integer> listb = new ArrayList<>();
listb.add(2);
listb.add(4);
List<List<Integer>> listc = new ArrayList<>();
listc.add(lista);
listc.add(listb);
System.out.println(listc);
List<Integer> listd = listc.stream().flatMap(ele -> ele.stream()).collect(Collectors.toList());
System.out.println(listd);

运行程序打印信息如下：

[[1, 3], [2, 4]]
[1, 3, 2, 4]

flatMap里面需要传入一个Function参数，这里根据需要将数组转为stream。其实可以直接传入一个函数名:Collection::stream，效果是一样的。

完整代码如下：

package org.example;import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;public class FlatMapExample {public static void main(String[] args) {List<Integer> lista = new ArrayList<>();lista.add(1);lista.add(3);List<Integer> listb = new ArrayList<>();listb.add(2);listb.add(4);List<List<Integer>> listc = new ArrayList<>();listc.add(lista);listc.add(listb);System.out.println(listc);List<Integer> listd = listc.stream().flatMap(Collection::stream).collect(Collectors.toList());System.out.println(listd);try {Stream<String> stream = Files.lines(Paths.get("test.txt"));// stream.forEach(System.out::println);List<String> words = stream.flatMap(line -> Stream.of(line.split(" +"))).collect(Collectors.toList());System.out.println(words);} catch (IOException e) {throw new RuntimeException(e);}}
}

运行截图：