IOS

关注公众号 jb51net

关闭
首页 > 软件编程 > IOS > Swift map 和 flatMap

Swift 进阶 —— map 和 flatMap的使用

作者:但江

这篇文章主要介绍了Swift map和flatMap的相关资料,帮助大家更好的理解和使用Swift,感兴趣的朋友可以了解下

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", "25", "New York"], ["Female", "18", "London"], ["Unkown", "Unkown", "Unkown"]]

输入是一个一维数组,转换后 infos2 的结果是如下的一个一维数组,所以 flatMap 后只有一层结构:

["Male", "25", "New York", "Female", "18", "London", "Unkown", "Unkown", "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?

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的资料请关注脚本之家其它相关文章!

您可能感兴趣的文章:
阅读全文