java8 映射方法（map，flatMap）

本文介绍: 一个非常常见的数据处理套路就是从某些对象中选择信息。比如在SQL里，你可以从表中选择一列。Stream API也通过map和flatMap方法提供了类似的工具。

一个非常常见的数据处理套路就是从某些对象中选择信息。比如在SQL里，你可以从表中选择一列。Stream API也通过map和flatMap方法提供了类似的工具。

流支持map方法，它会接受一个函数作为参数。这个函数会被应用到每个元素上，并将其映射成一个新的元素（使用映射一词，是因为它和转换类似，但其中的细微差别在于它是“创建一个新版本”而不是去“修改”）。

<R> Stream<R> map(Function<? super T, ? extends R> mapper);

List<Integer> dishNameLengths = menu.stream()
                                    .map(Dish::getName)
                                    .map(String::length)
                                    .collect(toList());

<R> Stream<R> flatMap(Function<? super T, ? extends Stream<? extends R>> mapper);

给定单词列表["Hello","World"]，你想要返回列表["H","e","l", "o","W","r","d"]。
List<String> uniqueCharacters = words.stream()
                                     .map(w -> w.split(""))      ←将每个单词转换为由其字母构成的数组
                                     .flatMap(Arrays::stream)    ←将各个生成流扁平化为单个流
                                     .distinct()
                                     .collect(Collectors.toList());

一言以蔽之，flatmap方法把一个流中的每个值都换成另一个流，然后把所有的流连接起来成为一个流。
在这里插入图片描述
为巩固对于map和flatMap的理解，如下测试：

给定[1, 2, 3, 4, 5]，应该返回[1, 4, 9, 16, 25]。

List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5);
List<Integer> squares = numbers.stream()
                               .map(n -> n * n)
                               .collect(toList());

例如，给定列表[1, 2, 3]和列表[3, 4]，应该返回[(1, 3), (1, 4), (2, 3), (2, 4), (3, 3), (3, 4)]。

List<Integer> numbers1 = Arrays.asList(1, 2, 3);
List<Integer> numbers2 = Arrays.asList(3, 4);
List<int[]> pairs = numbers1.stream()
                            .flatMap(i -> numbers2.stream().map(j -> new int[]{i, j}))
                            .collect(toList());

如何扩展前一个例子，只返回总和能被3整除的数对呢？例如(2, 4)和(3, 3)是可以的。

List<Integer> numbers1 = Arrays.asList(1, 2, 3);
List<Integer> numbers2 = Arrays.asList(3, 4);
List<int[]> pairs = numbers1.stream()
                            .flatMap(i ->numbers2.stream()
                                                 .filter(j -> (i + j) % 3 == 0)
                                                 .map(j -> new int[]{i, j}))
                            .collect(toList());

合并数组

# Map
Map<String, List<Person>> map = new HashMap<>();
List<Person> collect1 = map.values().stream().flatMap(Collection::stream).collect(Collectors.toList());

# List
List<Person> list1 = new ArrayList<>();
List<Person> list2 = new ArrayList<>();
# 方式1
List<Person> collect2 = Stream.of(list1, list2).flatMap(Collection::stream).collect(Collectors.toList());
# 方式2
List<Person> collect3 = Stream.of(list1.stream(), list2.stream()).flatMap(Function.identity()).collect(Collectors.toList());