Swift中的map和flatMap

在iOS 9以后的版本中,Swift语言中有了两个函数,一个叫做map, 一个叫做flatMap,这两个概念在我们的函数式编程中非常有用。到底这两个函数是什么意义呢?

map

首先看这个map,map顾名思义,就是映射,例如我们有一个数组,我们想把数组里面的每一个数乘以二。

let arr = [1, 2, 3, 4, 5, 6]
let twoTimesArray = arr.map { $0 * 2 }
twoTimesArray   // [2, 4, 6, 8, 10, 12]

从我们方法的定义可以看到,我们可以看到

Paste_Image.png

所以我们这个数组是不能是optional的,如果把上面的数组换成

let arr: [Int?] = [1, 2, 3, 4, 5, 6, nil]

我们就会报错了。

flatMap

概念

flatMap的主要作用,是把我们的这个把我们的多个集合合为一个集合。

let nestedArray = [[1, 2, 3], [4, 5, 6]]
let flattenedArray = nestedArray.flatMap{ $0 }
flattenedArray // [1, 2, 3, 4, 5, 6]

那么我们想让每个元素乘以2,试着做如下的代码

let nestedArray = [[1, 2, 3], [4, 5, 6]]
let flattenedArray = nestedArray.flatMap{ $0 * 2}
flattenedArray

什么?!竟然报错了,报错信息是这样的


Paste_Image.png

可以看到,错误提示了我们,实际上我们的的flatmap操作的不是数组中的元素,他操作的是整个数组。实际上,我们可以根据我们flatMap的定义也可以看到。第一个是这样:

Paste_Image.png

可以很清楚的看到,这个函数是用来连接我们的结果的,所以就是将我们两个数组连接起来。那么不仅如此,还有一个函数定义是这样的:

Paste_Image.png

这个函数是什么意思呢,他是用来把我们的nil的值去掉的,例如如下的例子

let optionalInts: [Int?] = [1, nil, 2, 3, nil, 4, nil, 5]
let ints = optionalInts.flatMap { $0}
ints // [1, 2, 3, 4, 5]

是把我们的数组中的nil去除掉,这是flatMap的另外一种用法。这个也会是我们开发中非常有用的一个功能。实际上,这两个方法是相似的,为什么说呢?

剖析

我们可以这么想,如果我们上面的例子,数组如果变成这种样子,我们可以得到

let nestedArray = [[],[1, 2, 3]]
let flattenedArray = nestedArray.flatMap{ $0}
flattenedArray // [1, 2, 3]

那么,我们的Optional其实也是一个枚举类型,这个可以从Swift的API就能看出,我们可以这样定义一个Optional.

let a: Optional<Int> = .Some(5)
let b: Optional<Int> = .None

所以我们可以把我们的数组中的nil看成是这种形式

let arr2: [Optional<Int>] = [.None, .Some(5), .None, .Some(2)]

这样就很容易理解,这两个函数的机制实际是一致的。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • map 和 flatMap 是 Swift 中两个常用的函数,它们体现了 Swift 中很多的特性。对于简单的使用...
    SwiftCafe阅读 1,586评论 5 16
  • 一、数组中的 map 和 flatMap 数组中的 map 对数组元素进行某种规则的转换,例如: 二、 flatM...
    kmplayer阅读 4,144评论 0 6
  • map .map操作,对数组中的元素添加相同类型值,返回相同类型 .map操作,对数组中的元素添加不同类型值,返回...
    Panco_Lee阅读 382评论 0 1
  • 粗鄙的简介 以下转自CSDN的一篇博博客http://blog.csdn.net/fish_yan_/articl...
    小白猿阅读 498评论 0 2
  • 看空某股票,大亏1.5w刀,又补了8k刀的看空,最近又亏了2k 简直了,明明知道大家都在看涨,自己也对这个行业抱有...
    下雨不愁_80阅读 196评论 0 0