map vs flatMap 一眼看懂
1️⃣ 核心概念
| 函数 | 功能 | 输出结果 |
|---|
map | 一对一映射 | 每个元素映射成一个新元素 |
flatMap | 一对多映射 + 扁平化 | 每个元素映射成一个集合/流,最终扁平成一个流 |
2️⃣ 通俗理解
- map:把每个元素“变成别的东西”,结果是原来的大小一样
比喻:每个人穿上不同衣服,人数不变
- flatMap:把每个元素“变成一个集合”,然后把所有集合合成一个大集合
比喻:每个人可能有多个爱好,把所有人的爱好合成一个列表
3️⃣ Java Stream 示例
示例数据
List<String> names = Arrays.asList("Alice", "Bob"); |
|---|
map 示例
List<Integer> nameLengths = names.stream() |
|---|
``.map(String::length) // 每个名字映射成长度 |
|---|
``.collect(Collectors.toList()); |
|---|
System.out.println(nameLengths); // [5, 3] |
|---|
- 输入:
["Alice", "Bob"]
- 输出:
[5, 3]
- 元素一对一映射
flatMap 示例
假设每个人有多个爱好:
Map<String, List<String>> hobbies = new HashMap<>(); |
|---|
hobbies.put("Alice", Arrays.asList("reading","swimming")); |
|---|
hobbies.put("Bob", Arrays.asList("running","gaming")); |
|---|
List<String> allHobbies = hobbies.values().stream() |
|---|
``.flatMap(List::stream) // 每个 List 扁平化成一个 Stream |
|---|
``.collect(Collectors.toList()); |
|---|
System.out.println(allHobbies); // [reading, swimming, running, gaming] |
|---|
- 输入:
[["reading","swimming"], ["running","gaming"]]
- 输出:
["reading","swimming","running","gaming"]
- 元素一对多映射 + 扁平化
4️⃣ 总结规律
| 方法 | 输入 | 映射结果 | 最终 Stream |
|---|
map | T | R | Stream |
flatMap | T | Stream | Stream(扁平化) |
map 得到的是“流里套流”,flatMap 得到的是“平铺的流”
5️⃣ 小技巧记忆法
- map = 一变一:元素个数不变
- flatMap = 扁平 + 展开:元素可以变多
- 思路:想象每个元素被打开,如果每个元素里还有多个东西,用
flatMap 扁平化
如果你愿意,我可以帮你画一张 map vs flatMap 流程图,直观展示:
``flatMap -> [结果1a, 结果1b, 结果2a, 结果2b] |
|---|
www.52runoob.com/archives/64…