快捷導(dǎo)航

Stream流中的Map與flatMap的區(qū)別及說明

更新時(shí)間：2025年09月19日 15:23:48 作者：night_gu

Map轉(zhuǎn)換單個(gè)元素為新值,保持流長(zhǎng)度；flatMap將元素轉(zhuǎn)換為流再合并,實(shí)現(xiàn)扁平化處理,兩者區(qū)別在于返回類型（對(duì)象vs流）及是否改變?cè)財(cái)?shù)量,適用于不同數(shù)據(jù)結(jié)構(gòu)轉(zhuǎn)換場(chǎng)景

Map與flatMap的基本概念

Map用于將流中的每個(gè)元素通過指定函數(shù)轉(zhuǎn)換為另一個(gè)元素，生成一個(gè)一對(duì)一映射的新流。

例如將字符串流轉(zhuǎn)換為對(duì)應(yīng)長(zhǎng)度流：

List<String> words = Arrays.asList("a", "bb", "ccc");
List<Integer> lengths = words.stream()
                             .map(String::length)
                             .collect(Collectors.toList()); 
// 輸出: [1, 2, 3]

flatMap用于將每個(gè)元素轉(zhuǎn)換為流后合并所有流，形成一對(duì)多映射的扁平化結(jié)果。

例如拆分字符串中的單詞：

List<String> lines = Arrays.asList("hello world", "java stream");
List<String> words = lines.stream()
                          .flatMap(line -> Arrays.stream(line.split(" ")))
                          .collect(Collectors.toList());
// 輸出: ["hello", "world", "java", "stream"]

核心區(qū)別對(duì)比

輸入輸出關(guān)系

Map保持原始流元素?cái)?shù)量不變，flatMap可能改變?cè)財(cái)?shù)量（如拆分為多個(gè)子流后合并）

返回值要求

Map的Function返回普通對(duì)象，flatMap的Function必須返回Stream對(duì)象

數(shù)據(jù)結(jié)構(gòu)處理

Map適合單一元素轉(zhuǎn)換，flatMap適合處理嵌套集合（如List<List>）

典型應(yīng)用場(chǎng)景

Map適用場(chǎng)景

數(shù)據(jù)類型轉(zhuǎn)換（String→Integer）、屬性提?。▽?duì)象→ID）、簡(jiǎn)單計(jì)算（數(shù)值→平方值）

flatMap適用場(chǎng)景

合并多個(gè)集合：List<List<Integer>> → List<Integer>
分解復(fù)合結(jié)構(gòu)：字符串句子→單詞流、樹節(jié)點(diǎn)→子樹節(jié)點(diǎn)流
過濾后展開操作：先filter再flatMap處理有效數(shù)據(jù)

性能注意事項(xiàng)

flatMap涉及流合并操作，可能產(chǎn)生更多臨時(shí)對(duì)象。

對(duì)于簡(jiǎn)單轉(zhuǎn)換優(yōu)先使用map，多層嵌套數(shù)據(jù)結(jié)構(gòu)才使用flatMap。

代碼示例對(duì)比

處理書籍作者場(chǎng)景：

// 使用map（保留嵌套結(jié)構(gòu)）
List<Book> books = ...;
List<List<String>> authors = books.stream()
                                 .map(Book::getAuthors)
                                 .collect(Collectors.toList());

// 使用flatMap（扁平化結(jié)構(gòu)）
List<String> allAuthors = books.stream()
                              .flatMap(book -> book.getAuthors().stream())
                              .collect(Collectors.toList());