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 操作是將流中的元素進(jìn)行再次加工形成一個新流。這在開發(fā)中很有用。比如我們有一個學(xué)生集合，我們需要從中提取學(xué)生的年齡以分析學(xué)生的年齡分布曲線。

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

示意圖：

對應(yīng)的偽代碼：

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

3. flatMap 操作

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

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

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

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

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

再畫一張圖來加深理解：

4. 總結(jié)

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

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

最新評論