Stream流中map与flatMap区别|转

发布时间 2023-07-15 21:45:21作者: 楼兰胡杨

  map和flatMap都是对流中的每一个元素,执行入参中的函数式方法,只是在函数式方法的返回值上有区别:

  • map中函数式方法返回值为对象,也就是原始流中有n个元素,处理之后的流中元素个数也是n。即使你在返回的时候返回了多个元素如数组,它也会把数组当成一个对象处理。
  • flatMap中函数式方法返回值为流,flat为扁平的意思,也就是说,我们可以通过flatMap改变原始流中对象的个数,flatMap会将返回的流合并成一个流,而不是当成对象处理。

  通过一个简单的例子来理解:假设有一个数组1,2,3,4,5,需要对其每个元素进行处理,当值为偶数时,则返回双个,也就是结果为:1,2,2,3,4,4,5,这个时候就需要使用flatMap来处理。我们可以通过下面的示意图及代码体验一下区别:

public class FlatMapTest {
    public static void main(String[] args) {
        List<Integer> nums = Arrays.asList(1, 2, 3, 4, 5);
        List<Object> res1 = nums.stream().map(e -> {
            if (e % 2 == 0) {
                return Arrays.asList(e, e);
            } else {
                return e;
            }
        }).collect(Collectors.toList());
        List<Object> res2 = nums.stream().flatMap(e -> {
            if (e % 2 == 0) {
                return Stream.of(e, e);
            } else {
                return Stream.of(e);
            }
        }).collect(Collectors.toList());
        System.out.println(res1);
        System.out.println(res2);
    }
}

执行结果:

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

Reference