java中flatMap用法

news2024/9/30 13:21:03

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);
        }
    }
}

运行截图：