arrays - 创建一个扩展以在 Swift 中从数组中过滤 nils

Question

我正在尝试编写 Array 的扩展，它将允许将可选 T 的数组转换为非可选 T 的数组。

例如这可以写成这样的自由函数:

func removeAllNils(array: [T?]) -> [T] {
    return array
        .filter({ $0 != nil })   // remove nils, still a [T?]
        .map({ $0! })            // convert each element from a T? to a T
}

但是，我不能让它作为扩展工作。我试图告诉编译器扩展仅适用于可选值数组。这是我目前所拥有的:

extension Array {
    func filterNils<U, T: Optional<U>>() -> [U] {
        return filter({ $0 != nil }).map({ $0! })
    }
}

(它不编译!)

Answer 1

从 Swift 2.0 开始，您无需编写自己的扩展来从数组中过滤 nil 值，您可以使用 flatMap，它会展平数组并过滤 nils:

let optionals : [String?] = ["a", "b", nil, "d"]
let nonOptionals = optionals.flatMap{$0}
print(nonOptionals)

打印:

[a, b, d]

注意事项:

有 2 个 flatMap 函数:

• 一个 flatMap 用于删除非零值，如上所示。引用 - https://developer.apple.com/documentation/swift/sequence/2907182-flatmap

• 另一个flatMap 用于连接结果。引用 - https://developer.apple.com/documentation/swift/sequence/2905332-flatmap