Swift 中map、flatMap、compactMap、filer、reduce的使用總結

Swift相比於Objective-C又一個重要的優點，它對函數式編程提供了很好的支持，Swift提供了map、flatMap、compactMap、filer、reduce這三個高階函數作為對容器的支持。

1.map ： 可以對數組中的每一個元素做一次處理

// 計算字符串的長度
let stringArray = ["Objective-C", "Swift", "HTML", "CSS", "JavaScript"]
func stringCount(string: String) -> Int {
    return string.count
}
stringArray.map(stringCount)
 
stringArray.map({string -> Int in
    return string.count
})
 
// $0代表數組中的每一個元素
stringArray.map{
    return $0.count
}

 

let arr = [1,2,3,4,5,6,7]
func addOne(n1: Int) ->Int{
    return n1+1;
}
let arr1 = arr.map(addOne)
print(arr1)//[2, 3, 4, 5, 6, 7, 8]

 

let arr = [1,2,3,4,5,6,7]
let arr1 = arr.map{(n1) in n1+1}
print(arr1) //[2, 3, 4, 5, 6, 7, 8]

 

let arr = [1,2,3,4,5,6,7]
let arr1 = arr.map{$0+1}
print(arr1) //[2, 3, 4, 5, 6, 7, 8]

 

 

 2.flatMap與map的區別

(1)flatMap返回后的數組中不存在nil, 同時它會把Optional解包

let array = ["Apple", "Orange", "Puple", ""]
 
let arr1 = array.map { a -> Int? in
    let length = a.count
    guard length > 0 else { return nil }
    return length  
}
arr1 // [5, 6, 5, nil]
 
let arr2 = array.flatMap { a-> Int? in
    let length = a.count
    guard length > 0 else { return nil}
    return length    
}    
arr2 // [5, 6, 5]

 （2）flatMap還能把多維數組變成一維數組：

let array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
 
let arr1 = array.map{ $0 }
arr1 // [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
 
let arr2 = array.flatMap{ $0 }
arr2 // [1, 2, 3, 4, 5, 6, 7, 8, 9]

 

（3）flatMap也能把兩個不同的數組合並成一個數組，這個合並的數組元素個數是前面兩個數組元素個數的乘積：

let fruits = ["Apple", "Orange", "Puple"]
let counts = [2, 3, 5]
 
let array = counts.flatMap { count in
    fruits.map ({ fruit in
         return fruit + "  \(count)"            
    })   
}
array // ["Apple 2", "Orange 2", "Puple 2", "Apple 3", "Orange 3", "Puple 3", "Apple 5", "Orange 5", "Puple 5"]

 

4、compactMap: 當后面的閉包中返回Optional的時候，在Swift 4.1之后應該使用compactMap代替map。
5、filer：過濾，可以對數組中的元素按照某種規則進行一次過濾：

// 篩選出字符串的長度小於10的字符串
let stringArray = ["Objective-C", "Swift", "HTML", "CSS", "JavaScript"]
func stringCountLess10(string: String) -> Bool {
    return string.count < 10
}
stringArray.filter(stringCountLess10)
 
stringArray.filter({string -> Bool in
    return string.count < 10
})
 
// $0表示數組中的每一個元素
stringArray.filter{
    return $0.count < 10
}

 

6、reduce：計算，可以對數組的元素進行計算：

// 將數組中的每個字符串用‘、’拼接
let stringArray = ["Objective-C", "Swift", "HTML", "CSS", "JavaScript"]
 
func appendString(string1: String, string2: String) -> String {
    return string1 == "" ? string2 : string1 + "、" + string2
}
// reduce方法中的第一個參數是初始值
stringArray.reduce("", appendString)
 
stringArray.reduce("", {(string1, string2) -> String in
    return string1 == "" ? string2 : string1 + "、" + string2
})
 
// $0表示計算后的結果, $1表示數組中的每一個元素
stringArray.reduce("", {
    return $0 == "" ? $1 : $0 + "、" + $1
})