java8 映射方法(map,flatMap)

5.2 映射(map,flatMap)

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

5.2.1 对流中每一个元素应用函数(map)

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

<R> Stream<R> map(Function<? super T, ? extends R> mapper);
List<Integer> dishNameLengths = menu.stream()
                                    .map(Dish::getName)
                                    .map(String::length)
                                    .collect(toList());

5.2.2 流的扁平化(flatMap)

<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());

-----------------------------------------------------------------------------读书笔记摘自 书名:Java 8实战 作者:[英] Raoul-Gabriel Urma [意] Mario Fusco [英] Alan M