目录
初探高阶函数
在 swift 中,高阶函数一共有下面几个:
- map:对给定数组每个元素,执行闭包中的映射,将映射结果放置在数组中返回。
- flatmap:对给定数组的每个元素,执行闭包中的映射,对映射结果进行合并操作,然后将合并操作后的结果放置在数组中返回。
- compactmap:对给定数组的每个元素,执行闭包中的映射,将非空的映射结果放置在数组中返回。
- compactmap对给定数组的每个元素,执行闭包中的映射,将非空的映射结果-键值对放置在字典中返回。
- filter:对给定数组的每个元素,执行闭包中的操作,将符合条件的元素放在数组中返回。
- reduce:对给定数组的每个元素,执行闭包中的操作对元素进行合并,并将合并结果返回。
通过上面的阐述,我们大概了解到了这几个函数是做什么用的,下面我们在通过几个例子来具体看一下代码上如何使用。
map
对于 map 函数,使用场景就是将数组的类型映射为别的类型。比如,我们有一个模型数组,模型的 id 字段我们从服务器拿的是 string 类型,在某种场景下我们需要转为 int 类型,这时候我们就可以通过 map 函数来实现该需求。
struct student { let id: string let name: string let age: int } let stu1 = student(id: "1001", name: "stu1", age: 12) let stu2 = student(id: "1002", name: "stu2", age: 14) let stu3 = student(id: "1003", name: "stu3", age: 16) let stu4 = student(id: "1004", name: "stu4", age: 20) let stus = [stu1, stu2, stu3, stu4] let intids = stus.map { (stu) in int(stu.id) } print(intids) //[optional(1001), optional(1002), optional(1003), optional(1004)]
通过上述代码,我们将 id 字段从 string 映射为了 int? 类型,这并不是我们想要的 int 类型。如果我们需要访问元素的话还得需要解包,那么我们如何既能将元素映射又能自动筛选 nil 的值呢?这时,就轮到 compactmap 出马了。
compactmap
我们将上面的代码替换为:
let intids = stus.compactmap { (stu) in int(stu.id) }
这时,我们再打印 intids 就会发现它已经为 int 类型了。
compactmapvalues
对于 set 和 array ,你可以使用 compactmap 来获得非空的集合,但是对于 dictionary 来说,这个函数是不起作用的。
let dict = ["key1": 10, "key2": nil] let result = dict.compactmap { $0 } print(result) //[(key: "key1", value: optional(10)), (key: "key2", value: nil)]
这时候,我们需要使用 compactmapvalues 函数来获得非空的字典。
print(result) //["key1": 10] let result = dict.compactmapvalues { $0 } print(result) //["key1": 10]
flatmap
对于 flatmap,主要的应用场景就是你想获得一个单层集合的数组。通过下面的代码来看一下 map 和 flapmap 的区别。
let scoresbyname = ["henk": [0, 5, 8], "john": [2, 5, 8]] let mapped = scoresbyname.map { $0.value } // [[2, 5, 8], [0, 5, 8]] print(mapped) let flatmapped = scoresbyname.flatmap { $0.value } // [2, 5, 8, 0, 5, 8]
map 会直接将元素放在数组中,而 flatmap 会将元素平铺在一个数组中。实际上,s.flatmap(transform) 等同于s.map(transform).joined()。
filter
这个函数就如同单词的意思:查找。将符合条件的元素查找出来放置在数组中返回。比如我们想查找年龄大于18岁的所有学生。
let adults = stus.filter { (stu) -> bool in stu.age >= 18 } print(adults) // 数组中只包含stu4 学生
reduce
对于 reduce,我们的使用场景就是对数组中的元素进行组合运算,比如我们想计算所有学生的年龄加载一起是多少。
let totalages = stus.reduce(0) { (result, stu) in return result stu.age } print(totalages) // 62
该函数的第一个参数为初始值,后面元组中的第一个参数为每次计算的结果,第二个参数为每次遍历的元素。最后将计算的结果返回。
组合使用
对于使用高阶函数最大的好处就是可以进行函数式编程了。下面我们通过几个小栗子来对这几个高阶函数进行组合使用。
将 string 类型映射为 int 类型,并查找id大于1002的所有学生
let adults = stus.compactmap { (stu) in int(stu.id) }.filter { (id) -> bool in id > 1002 } print(adults) //[1003, 1004]
计算年龄大于12的所有学生年龄总和
let totalage = stus.filter { (stu) -> bool in stu.age > 12 }.reduce(0) { (result, stu) in return result stu.age } print(totalage) // 50