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 5map是流的中间操作, 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算子就介绍完毕了!