前言
在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函数返回一个初始参数与原集合元素经过组合后的非集合类型的值。
这些集合操作可以说是我们日常编码中的利器,能够将一些很繁琐的操作简化提高效率。合适的使用将使我们在日常的编码工作中事半功倍。