Swift的flatMap的用法

2017-08-01

FlatMap

FlatMap是函数式编程中的另外一个用法，主要是用于处理多组数据合并一组，和归约是不同的

方法	作用
Reduce	一组数据里所有的元素合并成一个元素
FlatMap	多组数据里的所有元素合并成一组数据

Swift中的FlatMap

Swift中有两个FlatMap方法，可以接收两种不同的Lambda表达式

根据其描述不同

1
2

ElementOfResult: Returns an array containing the non-nil results of calling the given transformation with each element of this sequence.
Sequence: Returns an array containing the concatenated results of calling the given transformation with each element of this sequence.

讲人话就是

ElementOfResult不改变内部元素，仅仅删除掉nil的选项
Sequence把所有元素内部的元素也进行Flat(压平，合并到一起，concatenated)，并且不允许存在nil，必须把nil替换为默认值

let couple = [[1,2,3],[4,5,6], nil, [8,9]]

var nonnil = couple.flatMap({ (number) -> [Int]? in
    return number //返回元素[Int]，为Int数组，并且把nil过滤掉
})

var flatten = couple.flatMap({ (number) -> [Int] in
    return number ?? [7] //把所有元素[Int],Int数组合并成一个数组，并且把nil用[7]，仅有一个元素7的Int数组替换
})

print(nonnil) //[[1, 2, 3], [4, 5, 6], [8, 9]]
print(flatten) //[1, 2, 3, 4, 5, 6, 7, 8, 9]

更复杂的情况

如果在子元素的数组里也存在nil，则可以把两个flatMap方法结合起来用

let complex = [[1,2,3],[4,5,6], nil, [8,9,nil]]

let simple = complex.flatMap { (number) -> [Int] in
    //首先去除子数组的 [8,9,nil]里的nil
    let non_sub_nil = number?.flatMap({ (sub) -> Int? in
        return sub
    })
    //其次concatenated压平它们
    return non_sub_nil ?? [7]
}

print(simple) //[1, 2, 3, 4, 5, 6, 7, 8, 9]