问 在Swift中展平数组的数组

在Swift标准库中，为符合 Sequence 协议的所有类型(或Swift 3之前的 SequenceType 上的 flatten )实现了 joined 函数，其中包括 Array

let numbers = [[1,2,3],[4],[5,6,7,8,9]]
let flattened = Array(numbers.joined())

在某些情况下，使用 joined() 可能是有益的，因为它返回一个惰性集合，而不是一个新的数组，但在传递给 Array() 初始化程序时，总是可以转换为一个数组，就像上面的例子一样。

Swift 4.2

我在下面写了一个简单的数组扩展。可以使用展平包含另一个数组或元素的数组。与joined()方法不同。

public extension Array {
    public func flatten() -> [Element] {
        return Array.flatten(0, self)
    public static func flatten<Element>(_ index: Int, _ toFlat: [Element]) -> [Element] {
        guard index < toFlat.count else { return [] }
        var flatten: [Element] = []
        if let itemArr = toFlat[index] as? [Element] {
            flatten = flatten + itemArr.flatten()
        } else {
            flatten.append(toFlat[index])
        return flatten + Array.flatten(index + 1, toFlat)
}

用法：

let numbers: [Any] = [1, [2, "3"], 4, ["5", 6, 7], "8", [9, 10]]
numbers.flatten()

Apple Swift版本5.1.2 (swiftlang-1100.0.278 clang-1100.0.33.9)

目标: x86_64-apple-darwin19.2.0

​

​

let optionalNumbers = [[1, 2, 3, nil], nil, [4], [5, 6, 7, 8, 9]]
print(optionalNumbers.compactMap { $0 }) // [[Optional(1), Optional(2), Optional(3), nil], [Optional(4)], [Optional(5), Optional(6), Optional(7), Optional(8), Optional(9)]]
print(optionalNumbers.compactMap { $0 }.reduce([], +).map { $0 as? Int ?? nil }.compactMap{ $0 }) // [1, 2, 3, 4, 5, 6, 7, 8, 9]
print(optionalNumbers.compactMap { $0 }.flatMap { $0 }.map { $0 as? Int ?? nil }.compactMap{ $0 }) // [1, 2, 3, 4, 5, 6, 7, 8, 9]
print(Array(optionalNumbers.compactMap { $0 }.joined()).map { $0 as? Int ?? nil }.compactMap{ $0 }) // [1, 2, 3, 4, 5, 6, 7, 8, 9]