iOS开发 Swift 之 map 和 flatMap

需求场景：
有一个数组Array，该数组中存放了自定义的模型Model。由于产需求，现在需要将Model中的name属性统一拼接一个性别字符串，例如：name = "小黄"，将改为 name = "男同学"。

1.map函数的使用
let numbers = [1, 2, 3, 4, 5]
let result = numbers.map { $0 + 2 }
print(result) // 打印：[3, 4, 5, 6]

2.flatMap函数的使用
result = numbers.flatMap { $0 + 2 } // [3,4,5,6]
let numbersCompound = [[1, 2, 3], [4, 5, 6]]
var res = numbersCompound.map { $0.map{ $0 + 2 } } // [[3, 4, 5], [6, 7, 8]]
var flatRes = numbersCompound.flatMap { $0.map{ $0 + 2 } } // [3, 4, 5, 6, 7, 8]
***1.numbersCompound.map { … } 这个调用实际上是遍历了这里两个数组元素 [1,2,3] 和 [4,5,6]。 因为这两个元素依然是数组，所以我们可以对他们再次调用 map 函数：$0.map{ $0 + 2 }。 这个内部的调用最终将数组中所有的元素加 2。
***2.flatMap 依然会遍历数组的元素，并对这些元素执行闭包中定义的操作。 但唯一不同的是，它对最终的结果进行了所谓的 “降维” 操作。 本来原始数组是一个二维的， 但经过 flatMap 之后，它变成一维的了。

1.map的定义
func map (@noescape transform: (Self.Generator.Element) throws -> T) rethrows -> [T]
其中闭包为：(Self.Generator.Element) -> T ，Self.Generator.Element类型参数代表数组中当前元素的类型。而T这个返回值是可以和传递进来的值不同，比如我们可以这样：
let stringResult = numbers.map { "No. \($0)" } // // ["No. 1", "No. 2", "No. 3", "No. 4"]
这次我们在闭包装把传递进来的数字拼接到一个字符串中， 然后返回一个组数， 这个数组中包含的数据类型，就是我们拼接好的字符串。

2.flatMap的定义
func flatMap(transform: (Self.Generator.Element) throws -> T?) -> [T]
func flatMap(transform: (Self.Generator.Element) -> S) -> [S.Generator.Element]
和 map 不同， flatMap 有两个重载。 参照我们刚才的示例， 我们调用的其实是第二个重载
flatMap 的闭包接受的是数组的元素，但返回的是一个 SequenceType 类型，也就是另外一个数组。
我们传入给 flatMap 一个闭包$0.map{ $0 + 2 } , 这个闭包中，又对 $0 调用了 map 方法， 从 map 方法的定义中我们能够知道，它返回的还是一个集合类型，也就是 SequenceType。 所以我们这个 flatMap 的调用对应的就是第二个重载形式。

总结

map：是对原对象所有元素进行一对一转换处理，中间不会跳过或遗漏，包括nil元素，map 方法接受一个闭包作为参数， 然后它会遍历整个 numbers 数组，并对数组中每一个元素执行闭包中定义的操作。 相当于对数组中的所有元素做了一个映射。
flatMap：更灵活，可变换维度，也能够自动解包，所以当我们对不符合元素，返回nil，最终的结果是过滤掉nil的，从而能够实现过滤。

参考文章：
Swift中的map 和 flatMap 原理及用法