在Swift中展平数组数组

Question

Swift中是否有flattenScala,Xtend,Groovy,Ruby和co？

var aofa = [[1,2,3],[4],[5,6,7,8,9]]
aofa.flatten() // shall deliver [1,2,3,4,5,6,7,8,9] 

当然我可以使用reduce,但那有点糟透了

var flattened = aofa.reduce(Int[]()){
    a,i in var b : Int[] = a
    b.extend(i)
    return b
}

Answer 1

Swift 3.0

reduce:

let numbers = [[1,2,3],[4],[5,6,7,8,9]]
let reduced = numbers.reduce([], +)

flatMap:

let numbers = [[1,2,3],[4],[5,6,7,8,9]]
let flattened = numbers.flatMap { $0 }

joined:

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

Answer 2

在Swift标准库中,为joined符合Sequence协议的所有类型(或flatten在SequenceTypeSwift 3之前)实现了功能,其中包括Array:

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

在某些情况下,使用joined()可能是有益的,因为它返回一个惰性集合而不是一个新数组,但是在传递给Array()初始化器时总是可以转换为数组,如上例所示.

Answer 3

Swift 4.x

只是为了在数组中增加一点复杂性,如果有一个包含数组数组的数组,那么flatMap实际上会失败.

假设数组是

var array:[Any] = [1,2,[[3,4],[5,6,[7]]],8]

什么flatMap或compactMap回报是:

array.compactMap({$0})

//Output
[1, 2, [[3, 4], [5, 6, [7]]], 8]

为了解决这个问题,我们可以使用简单的for循环逻辑+递归

func flattenedArray(array:[Any]) -> [Int] {
    var myArray = [Int]()
    for element in array {
        if let element = element as? Int {
            myArray.append(element)
        }
        if let element = element as? [Any] {
            let result = flattenedArray(array: element)
            for i in result {
                myArray.append(i)
            }

        }
    }
    return myArray
}

所以用给定的数组调用这个函数

flattenedArray(array: array)

结果是:

[1, 2, 3, 4, 5, 6, 7, 8]

考虑到Int这里的情况,这个函数将有助于压扁任何类型的数组

游乐场输出:

Answer 4

斯威夫特 4.x

flatMap不推荐使用这种用法，它是为此而设计的。 https://developer.apple.com/documentation/swift/sequence/2905332-flatmap

var aofa = [[1,2,3],[4],[5,6,7,8,9]]
aofa.flatMap { $0 } //[1,2,3,4,5,6,7,8,9] 

Answer 5

编辑：joined()改为使用：

https://developer.apple.com/documentation/swift/sequence/2431985-joined

原回复：

let numbers = [[1, 2, 3], [4, 5, 6]]
let flattenNumbers = numbers.reduce([], combine: +)

Answer 6

斯威夫特 5.1

public extension Array where Element: Collection {

    func flatten() -> [Element.Element] {
        return reduce([], +)
    }
}

如果您还需要它作为字典值：

public extension Dictionary.Values where Value : Collection {
    func flatten() -> [Value.Element]{
         return self.reduce([], +)
    }
}