map 和 flatMap 主要分在集合上的使用和在可選型別上的使用，下面分別來看下。

集合上使用 map 和 flatMap

先看如下的程式碼：

func getInfos(by name: String) -> [String] {
 if name == "Jack" {
 return ["Male","25","New York"]
 } else if name == "Lucy" {
 return ["Female","18","London"]
 } else {
 return ["Unkown","Unkown","Unkown"]
 }
}

let names = ["Jack","Lucy","Nobody"]

let infos1 = names.map { getInfos(by: $0) }
print(infos1)

let infos2 = names.flatMap { getInfos(by: $0) }
print(infos2)

輸入是一個一維陣列，轉換後 infos1 的結果是如下的一個二維陣列，所以 map 後有兩層結構：

[["Male","New York"],["Female","London"],["Unkown","Unkown"]]

輸入是一個一維陣列，轉換後 infos2 的結果是如下的一個一維陣列，所以 flatMap 後只有一層結構：

["Male","New York","Female","London","Unkown"]

map 在 Array 上的實現大致如下：

extension Array {
 func map<T>(_ transform: (Element) -> T) -> [T] {
 var result: [T] = []
 for x in self {
  result.append(transform(x))
 }
 return result
 }
}

flatMap 在 Array 上的實現大致如下：

extension Array {
 func flatMap<T>(_ transform: (Element) -> [T]) -> [T] {
 var result: [T] = []
 for x in self {
  result.append(contentsOf: transform(x))
 }
 return result
 }
}

可選型別上使用 map 和 flatMap

如下程式碼中，輸入是 stringNumbers.first，其型別是 String?

• 轉換後 x 的型別是 Int??，所以 map 後有兩層 Optional
• 轉換後 y 的型別是 Int?，所以 flatMap 後只有一層 Optional

let stringNumbers = ["1","2","3","foo"]
let x = stringNumbers.first.map { Int($0) } // Optional(Optional(1))
let y = stringNumbers.first.flatMap { Int($0) } // Optional(1)

map 在 Optional 上的實現大致如下：

extension Optional {
 func map<U>(transform: (Wrapped) -> U) -> U? {
 if let value = self {
  return transform(value)
 }
 return nil
 }
}

flatMap 在 Optional 上的實現大致如下：

extension Optional {
 func flatMap<U>(transform: (Wrapped) -> U?) -> U? {
 if let value = self,let transformed = transform(value) {
  return transformed
 }
 return nil
 }
}

以上就是Swift 進階 - 詳解map 和 flatMap的詳細內容，更多關於Swift map 和 flatMap的資料請關注我們其它相關文章！