Swift 的集合類型Array,Dictionary中使用map,reduce,filter可能還不太習慣。除非你接觸過函數式語言,否則你的直覺是對for-in循環更加熟悉。本文主要介紹map,reduce,filter的使用。
Map
map函數循環集合將同樣的操作應用于集合的每一個元素。
簡單的map將整型數組中的每個數字都加10:
let data = [1, 2, 3, 4, 5]
let res = data.map { $0 + 10 }
這種腳手架的閉包語法一開始很難學習。map函數有一個參數是閉包在調用集合遍歷時返回的。閉包從集合中取一個元素作為參數然后返回結果。map函數在一個數組中返回所有結果。
??的函數表達式可以更清晰的寫成:
let res2 = data.map { (val: Int) -> Int in
return val + 10
}
閉包有一個單獨參數,但是swift能推斷類型,我們可以不用設置參數類型和return.
let res3 = data.map { val in val + 10 }
如果我們想寫的更簡潔,可以寫成第一種數字參數的形式~
返回的結果并不會顯示元素的類型,例如可以返回字符串類型:
let words = data.map { NumberFormatter.localizedString(from: $0 as NSNumber, number: .spellOut) }
print("\(words)")
["one", "two", "three", "four", "five"]
操作符不限制是數組類型,同樣可以用于Dictionary和Set類型,例如:
let dict = ["Fly": 80.0, "Elephant": 90, "FlyElephant": 100]
let dictResult = dict.map{ name,score in score * 0.5 }
print("\(dictResult)")
[50.0, 40.0, 45.0]
Filter
filter循環一個集合并返回符合包含條件的數組元素。
filter 函數有一個參數能夠傳入包含的條件。閉包每次都會從集合中取一個元素必須要返回一個bool值表明它是否應該包含在結果中。
let digits = [1,2,3,4]
let even = digits.filter { $0 % 2 == 0 }
print("\(even)")
[2, 4]
Reduce
reduce函數會拼接集合中所有的值來創建一個新的值。
let items = [2.0, 4.0, 6.0, 8.0]
let total = items.reduce(1.0, +)
print("\(total)")
21.0
字符串拼接:
let names = ["Fly", "Elephant", "FlyElephant"]
let text = names.reduce("", +)
print("\(text)")
FlyElephantFlyElephant
拼接參數是一個閉包同樣可以使用尾閉包的寫法:
let tails = names.reduce("~~~") {text, name in "\(text),\(name)"}
print("\(tails)")
~~~,Fly,Elephant,FlyElephant
總結
下次遇到需要遍歷集合類型的時候可以通過以下幾點進行map,filter和reduce判斷。
- map返回一個包含每個元素都進行轉變的數組;
- fiter返回一個僅僅符合條件的數據的數組;
- reduce返回單個值,值是通過每個元素的聯合閉包和一個初始值計算而成;
參考鏈接
https://useyourloaf.com/blog/swift-guide-to-map-filter-reduce/