java8中stream的map和flatmap的理解

转自https://blog.csdn.net/wynjauu/article/details/78741093

假如我们有这样一个需求给定单词列表["Hello","World"]，你想要返回列表["H","e","l", "o","W","r","d"],
对于这样的需求，我们可能想到的第一个版本可能是这样子的：

words.stream()
.map(word -> word.split(""))
.distinct()
.collect(toList());

这个方法的问题在于，传递给map方法的Lambda为每个单词返回了一个String[]（String
列表）。因此， map 返回的流实际上是Stream<String[]> 类型的。你真正想要的是用
Stream<String>来表示一个字符流。因此，这是行不通的。

以下是我对这个问题的解法和分步写法，希望能对你有帮助：

对flatMap的说明：这个在这里的主要作用是对流进行扁平化

String ss = "Hello";

          String[] aa = ss.split("");

          String[] bb = {"H", "e", "l", "l", "o"};

          String[] strings = {"Hello", "World"};

         //Arrays.stream接收一个数组返回一个流
         List<Stream<String>> streamList = Arrays.asList(strings).stream().
                 map(str -> str.split("")).
                 map(str -> Arrays.stream(str)).
                 collect(Collectors.toList());

         //分步写(map)

         Stream<String[]> stream = Arrays.asList(strings).stream().
                 map(str -> str.split(""));

         Stream<Stream<String>> streamStream = stream.map(strings1 -> Arrays.stream(strings1));
         List<Stream<String>> streamList1 = streamStream.collect(Collectors.toList());

         List<String> stringList = Arrays.asList(strings).stream().
                 map(str -> str.split("")).
                 flatMap(str -> Arrays.stream(str))
                 .collect(Collectors.toList());

         //分步写(流只能消费一次)(flatMap)
         Stream<String[]> stream1 = Arrays.asList(strings).stream().
                 map(str -> str.split(""));

         Stream<String> stringStream = stream1.flatMap(strings1 -> Arrays.stream(strings1));

         List<String> stringList1 = stringStream.collect(Collectors.toList());

另一篇转自：jdk8 Stream map和flatMap区别

• 1.map

• 将一种类型的值转换为另外一种类型的值。
• 代码：将List 转换成List

 List<String> collected = Stream.of("a", "b").collect(Collectors.toList());

  List<Integer> figure = collected.stream().map(s -> {
            Integer i;
            switch (s) {
                case "a":
                    i = 1;
                    break;
                case "b":
                    i = 2;
                    break;
                default:
                    i = -1;
                    break;
            }
            return i;
        }).collect(Collectors.toList());

• 2.flatMap（类似C# AddRange）

• 将多个Stream连接成一个Stream，这时候不是用新值取代Stream的值，与map有所区别，这是重新生成一个Stream对象取而代之。

   List<Integer> a=new ArrayList<>();
        a.add(1);
        a.add(2);
        List<Integer> b=new ArrayList<>();
        b.add(3);
        b.add(4);
        List<Integer> figures=Stream.of(a,b).flatMap(u->u.stream()).collect(Collectors.toList());
        figures.forEach(f->System.out.println(f));

• 或者看下面这个例子

• flatMap方法
  [java] view plain copy

  1. List<String> words = new ArrayList<String>();
  2. words.add("your");
  3. words.add("name");
  4. 
     
  5. public static Stream<Character> characterStream(String s){
  6. List<Character> result = new ArrayList<>();
  7. for (char c : s.toCharArray())
  8. result.add(c);
  9. return result.stream();
  10. }
  11. Stream<Stream<Character>> result = words.map(w -> characterStream(w));
  12. Stream<Character> letters = words.flatMap(w -> characterStream(w));

  如果使用的是map方法，返回的是[ ...['y', 'o', 'u', 'r'], ['n', 'a', 'm', 'e']]

  如果使用的是flatMap方法，返回的是['y', 'o', 'u', 'r', 'n', 'a', 'm', 'e']

  这是map和flatMap的区别

•     @Test
          public void testMapAndFlatMap() {
              List<String> words = new ArrayList<String>();
              words.add("hello");
              words.add("word");  
  
              //将words数组中的元素再按照字符拆分，然后字符去重，最终达到["h", "e", "l", "o", "w", "r", "d"]
              //如果使用map，是达不到直接转化成List<String>的结果
              List<String> stringList = words.stream()
                      .flatMap(word -> Arrays.stream(word.split("")))
                      .distinct()
                      .collect(Collectors.toList());
              stringList.forEach(e -> System.out.println(e));
          }