1.前言

Java 8提供了非常好用的 Stream API ,可以很方便的操作集合。今天我們來探討兩個 Stream 中間操作 map(Function<? super T, ? extends R> mapper) 和 flatMap(Function<? super T, ? extends Stream<? extends R>> mapper)

2. map 操作

map 操作是將流中的元素進行再次加工形成一個新流。這在開發(fā)中很有用。比如我們有一個學生集合，我們需要從中提取學生的年齡以分析學生的年齡分布曲線。

放在 Java 8 之前 我們要通過新建一個集合然后通過遍歷學生集合來消費元素中的年齡屬性?，F(xiàn)在我們通過很簡單的流式操作就完成了這個需求。

示意圖：

對應的偽代碼：

// 偽代碼
 List<Integer> ages=studentList.stream().map(Student::getAge).collect(Collectors.toList());

3. flatMap 操作

通過上面的例子， map 操作應該非常好理解。那么 flatMap 是干嘛的呢？ 這樣我們把上面的例子給改一下，如果是以班級為單位，提取所有班級下的所有學生的年齡以分析學生的年齡分布曲線。這時我們使用上面的方法還行得通嗎？

List<List<Student>> studentGroup= gradeList.stream().map(Grade::getStudents).collect(Collectors.toList());

通過上面的一頓操作，我們只能得到每個班的學生集合的集合 List<List<Student>> 。 我們還需要嵌套循環(huán)才能獲取學生的年齡數(shù)據(jù)，十分不便。如果我們能返回全部學生的集合 List<Students> 就方便多了。 沒錯！ flatMap 可以搞定！

// flatMap 提取 List<Students> map 提取年齡
 List<Integer> ages = grades.stream().flatMap(grade -> grade.getStudents().stream()).map(Student::getAge).collect(Collectors.toList());

正如上面的偽代碼所示，我們使用 flatMap 將所有的學生匯聚到一起。然后再使用 map 操作提取年齡。 flatMap 不同于 map 地方在于 map 只是提取屬性放入流中，而 flatMap 先提取屬性放入一個比較小的流，然后再將所有的流合并為一個流。有一種 “聚沙成塔” 的感覺。

再畫一張圖來加深理解：

4. 總結

map 操作和 flatMap 操作一旦你熟悉了，可以非常簡便地解決一些數(shù)據(jù)流的操作問題。擴展一下知識，其實Java 8 中 不光 Stream 中存在這兩種操作，其實 Optional<T> 中也存在這兩種操作，作用都差不多。

以上所述是小編給大家介紹的Java 8 Stream Api 中的 map和 flatMap 操作方法，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉載，煩請注明出處，謝謝！