【Java 8】Stream中flatMap方法


在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]

這里是從一維升為二維,邏輯如下:
在這里插入圖片描述

總結:flatMapmap結果歸類,從而達到升降維目的


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM