Swift 中 map 和 flatMap 的区别 (入门版)

在 Swift 中，map 和 flatMap 是用于处理集合类型（如数组、字典、集合等）的两个重要方法。尽管它们看起来相似，但它们在处理集合中的元素和结果时有着不同的行为。

map 方法

map 方法会对集合中的每一个元素应用一个变换（transform）函数，并返回一个包含所有变换后的新集合。结果集合的结构与原集合保持一致。

示例：

let numbers = [1, 2, 3, 4, 5]
let squares = numbers.map { $0 * 2 }
print(squares) // 输出: [2, 4, 6, 8, 10]

在这个例子中，每个数字都乘以 2，然后 map 返回一个包含这些结果的新数组。

flatMap 方法

flatMap 方法首先会对集合中的每一个元素应用一个变换函数，然后将所有结果“展平”到一个新的集合中。

示例：

let nestedNumbers = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
let flatNumbers = nestedNumbers.flatMap { $0 }
print(flatNumbers) // 输出: [1, 2, 3, 4, 5, 6, 7, 8, 9]

在这个例子中，flatMap 首先应用变换函数，将每个子数组提取出来，然后将所有子数组展平为一个单一的数组。

处理可选类型（Optional）

flatMap 在处理可选类型时非常有用，它会自动解包可选值并过滤掉 nil 值，而 map 则会保留 nil。

示例：

let numbers: [Int?] = [1, 2, nil, 4, nil, 5]
let mappedNumbers = numbers.map { $0 }
print(mappedNumbers) // 输出: [Optional(1), Optional(2), nil, Optional(4), nil, Optional(5)]

let flatMappedNumbers = numbers.flatMap { $0 }
print(flatMappedNumbers) // 输出: [1, 2, 4, 5]

在这个例子中，flatMap 会解包可选值并移除 nil 值，而 map 会保留 nil 值。

总结

• map：对每个元素应用变换函数，返回一个与原集合结构相同的新集合。
• flatMap：对每个元素应用变换函数，然后将结果展平成一个新的集合，特别适用于处理嵌套集合或可选类型。

选择使用 map 还是 flatMap 取决于具体需求。如果只需要对集合中的元素进行变换并保留原结构，使用 map；如果需要展平嵌套集合或处理可选类型，使用 flatMap。