在2D数组swift中访问相同的索引元素

Question

在Swift中是否存在一种语法方式来访问2D数组中数组内的相同索引元素，类似于访问表中的列，例如

let A = [ [ 1, 2, 3 ], [ 4, 5, 6 ], [ 7, 8, 9 ] ]

A[ 2 ]

得出[7,8,9]。有没有办法做类似

的事情

A[][ 2 ]

产量[3,6,9]？

我知道我能做到：

var B = [ Int ]()
for c in A {
    B.append( c[ 2 ] )
}

但我想知道是否有另一种，或许是语法的方式。

Answer 1

试试这个：

A.map { (elem) -> Int in
    return elem[2]
}

或简短版本（归功于@Ian）：

A.map { $0[2] }