Swift - 高阶函数map、flatMap、filter、reduce

Swift 提供了如下几个高阶函数:mapflatMapfilterreduce。使用高阶函数进行函数式编程不仅可以简化我们的代码,而且当数据比较大的时候,高阶函数会比传统实现更快。

一、map 函数

map 方法获取一个闭包表达式作为其唯一参数。然后它会遍历整个数组,并对数组中每一个元素执行闭包中的操作,并返回该元素所映射的值。
简单说就是,数组中每个元素通过某个方法进行转换,最后返回一个的数组。

示例1:将数组中的数字,全部转换为字符串

let prices = [1, 2, 3, 4]
let priceStr = prices.map { (n) -> String in
    return "价格:\(n)"
}
print(priceStr) // ["价格:1", "价格:2", "价格:3", "价格:4"]

注:在闭包中,可以用$0表示第 0 个参数,$1表示第 1 个参数,以此类推;若返回值只有一句,则可以省略return,所以上面的代码可以简化为:

let prices = [1, 2, 3, 4]
let priceStr = prices.map { "价格\($0)" }
print(priceStr) // ["价格:1", "价格:2", "价格:3", "价格:4"]

示例2:对一个数组里面的数据进行平方操作

let values = [5, 6, 7]
let squares = values.map { $0 * $0 }
print(squares) // [25, 36, 49]

二 flatMap 函数

flatMap 方法同 map 方法比较类似,只不过它返回后的数组中不存在 nil(自动把 nil 给剔除掉),同时它会把 Optional 解包。

// 示例1:
let names = ["Jack", "Rose", nil, "AnyBody"]
let names1 = names.flatMap { $0 }
print(names1) // 输出时,会自动剔除nil,结果:["Jack", "Rose", "AnyBody"]

// 示例2:
let names = ["Jack", "Rose", "AnyBody"]
let names2 = names.flatMap { (name) -> String? in
    return name
}
print(names2) // 结果与示例1相同

// 示例3:
let names = ["Jack", "Rose", "AnyBody"]
let names3 = names.flatMap { (name) -> String in
    return name
}
print(names3) // 结果: ["J", "a", "c", "k", "R", "o", "s", "e", "A", "n", "y", "B", "o", "d", "y"]

特别注意:如果 faltMap 闭包的返回值不是可选型(示例3),可能会有意想不到的结果。

同时,flatMap 还能把数组中存有数组的数组(二维数组、N维数组)一同打开变成一个新的数组。

let array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
let arr1 = array.map { $0 }   // [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
let arr2 = array.flatMap { $0 } // [1, 2, 3, 4, 5, 6, 7, 8, 9]

小结:mapflatMap 区别,map 函数值对元素进行变换操作。 但不会对数组的结构造成影响。 而 flatMap 会影响数组的结构。

在 Swift 4 时,flatMap函数过时了,取而代之的是compactMap(_:),其用法与flatMap一致,只是改了名字而已,不再赘述。

三、filter 函数

filter 方法用于过滤元素,即筛选出数组元素中满足某种条件的元素。

// 筛选出金额大于 25 的元素。
let prices = [20, 30, 40]
let result = prices.filter { $0 > 25 }
print(result) // [30, 40]

四、reduce 函数

reduce 方法把数组元素组合计算为一个值,并且会接受一个初始值,这个初始值的类型可以和数组元素类型不同。

示例1:计算数组元素的乘积

let prices = [20, 30, 40]
let result = prices.reduce(1) { $0 * $1 }
print(result) // 24000

上面的方法的第二行还可以简写成:

let result = prices.reduce(1, *)

示例2:将数组转成字符串,每个元素用顿号隔开。

let array = ["Apple", "Orange", "Grape"]
let str = array.reduce("") { $0 == "" ? $1 : $0 + "、" + $1 }
print(str) // Apple、Orange、Grape

也可以简写为:

let str = array.joined(separator: "、") // Apple、Orange、Grape
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,716评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,558评论 1 294
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,431评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,127评论 0 209
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,511评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,692评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,915评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,664评论 0 202
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,412评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,616评论 2 245
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,105评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,424评论 2 254
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,098评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,096评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,869评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,748评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,641评论 2 271

推荐阅读更多精彩内容