在java 8 Stream中,flatMap方法是一個維度升降
的方法
舉例說明
給 定 單 詞 列 表[“Hello”,“World”] ,要返回列表 [“H”,“e”,“l”, “o”,“W”,“r”,“d”] 。
使用map
方法,代碼如下:
public class StreamStr {
public static void main(String[] args) {
List<String> list = Arrays.asList("tom", "jame", "jerry", "hello");
Stream<String> stream = list.stream();
Stream<String[]> streamString = stream.map(s->s.split(""));
Stream<Stream<String>> map = streamString.map(Arrays::stream);
// List<Stream<String>> collect = map.collect(Collectors.toList());
// System.out.println(collect);
map.forEach(x->{
x.forEach(s->{
System.out.println(s);
});
});
}
}
t
o
m
j
a
m
e
j
e
r
r
y
h
e
l
l
o
轉變類型為:String
-> String[]
-> Stream<String>
,當前維度從一維變成二維,map
本身不能降為,執行如下
引入flatMap方法
具體代碼變化如下:
public class StreamStr {
public static void main(String[] args) {
List<String> list = Arrays.asList("tom", "jame", "jerry", "hello");
Stream<String> stream = list.stream();
Stream<String[]> streamString = stream.map(s->s.split(""));
Stream<String> map = streamString.flatMap(Arrays::stream);
map.forEach(x->{
System.out.println(x);
});
}
}
t
o
m
j
a
m
e
j
e
r
r
y
h
e
l
l
o
這里是從二維降為一維,邏輯如下:
flatMap升維
給定兩個數字列表,如何返回所有的數對呢?例如,給定列表[1, 2, 3]和列表[3, 4],應該返回[(1, 3), (1, 4), (2, 3), (2, 4), (3, 3), (3, 4)]。
public class StreamInt {
public static void main(String[] args) {
List<Integer> numbers1 = Arrays.asList(1, 2, 3);
List<Integer> numbers2 = Arrays.asList(3, 4);
// flatMap升維度
List<int[]> pairs = numbers1.stream().flatMap(x -> numbers2.stream().map(y -> new int[] { x, y }))
.collect(Collectors.toList());
for (int[] pair : pairs) {
System.out.println(Arrays.toString(pair));
}
}
}
測試結果:
[1, 3]
[1, 4]
[2, 3]
[2, 4]
[3, 3]
[3, 4]
這里是從一維升為二維,邏輯如下:
總結:flatMap
將map
結果歸類,從而達到升降維目的