map 和 flatMap 的区别

_板栗_2023-12-03 14:28

StreamAPI 中的 map 方法，实现是一种映射关系。类似数学中的 x -> f(x)。

而 flatMap 方法是在 map 的基础上做了一次 flat。

相信看了下面的例子大家会有更加直观的理解：

java 复制代码

public static void main(String[] args) {
        List<Integer> list1 = Arrays.asList(1, 2, 3);
        List<Integer> list2 = Arrays.asList(4, 5, 6);
        List<Integer> list3 = Arrays.asList(7, 8, 9);

        List<List<Integer>> listOfList = new ArrayList<>();
        listOfList.add(list1);
        listOfList.add(list2);
        listOfList.add(list3);

        System.out.println("listOfList: " + listOfList);

        // map 实现给每个元素加 10，但还是保留原有的 List<List<Integer>> 结构
        List<List<Integer>> collect1 = listOfList.stream()
                .map(list -> list.stream().map(e -> e + 10).collect(Collectors.toList()))
                .collect(Collectors.toList());
        System.out.println("collect1: " + collect1);

        // flatMap 实现给每个元素加 10，并 flat 成 List<Integer> 结构
        List<Integer> collect2 = listOfList.stream().flatMap(list -> list.stream().map(e -> e + 10))
                .collect(Collectors.toList());
        System.out.println("collect2: " + collect2);
    }

输出结果：

shell 复制代码

listOfList: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
collect1: [[11, 12, 13], [14, 15, 16], [17, 18, 19]]
collect2: [11, 12, 13, 14, 15, 16, 17, 18, 19]