Swift集合类型操作指南:Map、Filter、Reduce

前言

在OC中操作集合类型往往要使用for - in,但在Swift中,Apple为我们准备了Map、Filter、Reduce等函数,为我们在操作集合类型时候提供了许多便利。

Map、FlatMap

Map函数的操作是:使用 map 来遍历集合并对集合中每一个元素进行同样的操作。

这里需要注意的是,Map函数的返回值是一个数组。

现在我们有一个数组 testArr

let testArr = [10, 20, 30, 40, 50]

现在我们需要让数组每个元素都乘以一个自己。
for - in 写法如下:

var squareArr : [Int] = []
for value in testArr {
    squareArr.append(value * value)
}
//squareArr:[100, 400, 900, 1600, 2500]

这样写的方法是没毛病的,但是通过Map函数我们可以将代码精简:

let squareArr = testArr.map { (value : Int) -> Int in
    return value * value
}
//squareArr:[100, 400, 900, 1600, 2500]

这样写的好处是显而易见的,我们再也不需要for - in循环,在Map函数中Swift已经自动为我们做好了这一切,同时squareArr是不可变数组,并且我们不需要指定类型,Swift能够自动推断。

代码中我们可以看到map函数只有一个闭包参数,并且闭包中的参数value也能自动推断,于是我们可以省略一部分代码

let squareArr = testArr.map{value in value * value}

in的作用是将闭包参数和闭包执行代码区隔开。同时,我们也可以通过$来简写。
用$0、$1、$2来表示调用闭包中参数,$0指代第一个参数,$1指代第二个参数,$2指代第三个参数,以此类推$n+1指代第n个参数,$后的数字代表参数的位置,一一对应。

let squareArr = testArr.map{$0 * $0}

返回值的类型不受限于原集合中元素的类型
如,还是原来的数组,将整型映射至字符串的操作:

let words = testArr.map {NumberFormatter.localizedString(from: NSNumber.init(value: $0), number: .spellOut)}
//words :["ten", "twenty", "thirty", "forty", "fifty"]

map函数可以作用于任何集合类型,并不局限于数组,Dictionary或者Set都是可以的
如:

let userDic = ["a":1.6, "b":2, "c":1.8]
let nameArr = userDic.map {name,value in name}
//nameArr :["b", "a", "c"]

let lengthInMeters: Set = [4.0,6.2,8.9]
let lengthInFeet = lengthInMeters.map {meters in meters * 3}
FlatMap

FlatMap的作用是将一个多维的数组展开,组合成一个数组。
也可用于将集合中空值移除的方法

如:

let values = [ [3,2,1], [4,5], [6] ]
let flats = values.flatMap{ $0 }
//flats :  [3, 2, 1, 4, 5, 6]

let possibleNumbers = ["1", "2", "three", "///4///", "5"]
let mapped: [Int?] = possibleNumbers.map { str in Int(str) }
// [1, 2, nil, nil, 5]
let flatMapped: [Int] = possibleNumbers.flatMap { str in Int(str) }
// [1, 2, 5]

Map函数和FlatMap函数结合,能够做到许多复杂的操作,省去许多麻烦。

如:求二维数组每个元素平方并转化成String输出结果

let collection  = [[3,2,1],[4,5],[6]]
let resutl = collection.flatMap{
    values in values.map{
        number in String(number * number)
    }
}
//["9", "4", "1", "16", "25", "36"]

同时也可以简化成:

let collection  = [[3,2,1],[4,5],[6]]
let result = collection.flatMap{ $0.map{ String($0 * $0)} }

Filter

filter函数会遍历一个集合,并返回一个 Array,其中包含了集合中满足过滤条件的元素。

如:返回数组内偶数的集合

let numbers = [1,2,3,4,5,6,7,8]
let result = numbers.filter { (value) -> Bool in
    return value % 2 == 0
}
//[2, 4, 6, 8]

同时也可以简写成

let numbers = [1,2,3,4,5,6,7,8]
let result = numbers.filter{ $0 % 2 == 0}

Reduce

使用reduce来组合集合中的所有元素并返回一个非集合类型的值。
Reduce与上面的方法最大的区别就是返回值是一个非集合类型的值。
Reduce方法第一个参数是结果的初始值,第二个参数是个闭包函数。
如:返回数组中所有元素之和

let numbers = [1, 2, 3, 4]
let numberSum = numbers.reduce(0) { (x, y)  in
    x + y
}

同样可以简化为:

let numbers = [1,2,3,4,5,6,7,8]
let result = numbers.reduce(0, + )

值得一提的是,+运算同样支持字符串操作

let words = ["fu","ck","ing"," com","ing"]
let result = words.reduce("", +)
//fucking coming

应用题

例题:求数组中所有大于10的偶数之和。

let numbers = [12, 23, 99, 1, 0, 45, 2, 444, 424, 17, 80, 9]
let result = numbers.filter{$0 > 10 && $0%2 == 0}.reduce(0, +)
//960

总结

map函数返回一个包含了对原集合中每一个元素经过映射后的Array。

filter函数返回一个包含原集合中满足筛选条件的元素的Array。

reduce函数返回一个初始参数与原集合元素经过组合后的非集合类型的值。

这些集合操作可以说是我们日常编码中的利器,能够将一些很繁琐的操作简化提高效率。合适的使用将使我们在日常的编码工作中事半功倍。

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

推荐阅读更多精彩内容