Java流中的map算子和flatMap算子的区别

• 2020 年 3 月 28 日
• 筆記

map算子和flatMap算子

map和flatMap都是映射（转换），那么他们之间究竟有什么区别呢？

1.我们先简单了解下map算子：

 @org.junit.Test   public void test1(){  	List<String> words = Arrays.asList("hello","world");  	words.stream()  	.map(String::length)  //使用了方法引用，将String类型转换为int类型  	.forEach(System.out::println);   }  //输出：5   5  

map是流的中间操作， Stream map(Function<? super T, ? extends R> mapper)

传入一个Function函数式接口，返回一个流。关于函数式接口和lambda表达式可以看我之前的随笔。

2.再看个例子，来简单了解下两者间的区别，先着重思考一下下面的案例用map会有什么问题？

​ 需求：传入一个集合 list = Arrays.asList("hello","Hello")

​ 要求：输出：h,e,l,o,H （单词里面的每一个字母去重 ，并且进行打印）

 @org.junit.Test   public void test2(){  //错误演示  	 List<String> list = Arrays.asList("hello","Hello");//h,e,l,o,H  	 list.stream()  	 .map(m->m.split(""))  	 .distinct()  	 .collect(toList())  	 .stream()  	 .forEach(System.out::println);   }  

那么输出结果符合我们的需求吗？

//输出结果：这俩行是个什么鬼？  [Ljava.lang.String;@51e2adc7  [Ljava.lang.String;@1a8a8f7c  

这显然不符合我们的要求，那究竟是哪里出错了呢？我们一步一步分析一下：

1.Stream baseRDD = list.stream(); 将集合转换为流

2.Stream<String[]> mapRDD= baseRDD.map(w->w.split("")); 调用map算子，转换

3.Stream<String[]> distinctRDD= mapRDD.distinct(); 调用distinct算子，去重

4.List<String[]> collectRDD = distinctRDD.collect(toList()); 流转换为集合，最后输出

我们一步一步分析下来，看起来是不是还是没问题呢？接下来让我们再改写一下：

@org.junit.Test  public void test3(){    	List<String> words =  Arrays.asList("hello","Hello");//h,e,l,o,H  	Stream<String[]> mapRDD =  words.stream()  	.map(word->word.split(""));  	mapRDD.map(Arrays::stream)  	.distinct()  	.collect(toList())  	.stream()  	.forEach(System.out::println);    }  

//输出：这又是个什么鬼？？？是不是越来越迷糊了，不要着急  java.util.stream.ReferencePipeline$Head@2353b3e6  java.util.stream.ReferencePipeline$Head@631330c  

再来一步一步分析一下：

Stream<String[]> map1RDD = words.stream().map(word->word.split("")); 转换

Stream<Stream> map2RDD = map1RDD.map(d->Arrays.stream(d));

Stream<Stream> distinctRDD = map2RDD.distinct();

List<Stream> finalList = distinctRDD.collect(toList());

这里我们需要注意的是：map转化（输入一个类型，返回另外一个类型）

​ 输入类型的值（值的类型是上一个数据集的泛型）

​ 输出数据的类型会作为下一个数据集的泛型

还是不理解？没关系，等会我们通过图来深刻理解一下。

那flatMap算子呢？

@org.junit.Test  public void test4(){  	List<String> words =  Arrays.asList("hello","Hello");//h,e,l,o,H  	words.stream()  	.map(word->word.split(""))  	.flatMap(d->{  		      Stream<String> stream = Arrays.stream(d);  		return Arrays.stream(d);  	})  	.distinct()  	.forEach(System.out::println);  }  

//输出  heloH  

居然成功实现了需求，这两者究竟有什么区别呢？我们通过图画来更加生动形象的了解下（画图不易，值得一赞）

对比一下flatMap算子：

这样就一目了然了！！！

总结一下：

​ map算子：map返回值类型就是新的数据集的泛型

​ flatMap算子: flatMap算子返回类型就是新的数据集的类型

最后再看个案例：

​ 需求：输入：【1，2，3】【3，4】

​ 输出：【（1，3），（1，4），（2，3），（2，4），（3，3），（3，4）】

@org.junit.Test  public void test7(){  	//集合（.stream()）  	List<Integer> number1 = Arrays.asList(1,2,3);  	List<Integer> number2 = Arrays.asList(4,5);  	/**  	 * 倒着推理一下 ：  	 *   最终的结果（终端）： List<int[]>  	 *   终端操作的上一个操作：Stream<int[]>  	 *  	 */      List<int[]> pairs = number1.stream()      .flatMap(i->{  //	    	 Stream<int[]> stream =  //	    			 number2.stream().map(j->new int[]{i,j});      	return number2.stream().map(j->new int[]{i,j});      })      .collect(toList());        Stream<int[]> intArrayRDD =  pairs.stream();      intArrayRDD.forEach(arr->{      	for(int i : arr){      		System.out.print(i+" ");      	}      	System.out.println();      });  }//OK  

至此，map算子和flatMap算子就介绍完毕了！