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