一个非常常见的数据处理套路就是 从某些对象中选择信息。比如在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());
-----------------------------------------------------------------------------读书笔记摘自 书名:Java 8实战 作者:[英] Raoul-Gabriel Urma [意] Mario Fusco [英] Alan M