Scala 中高效的二维数组列提取

Question

以二维数组为例

scala> val a = Array.tabulate(2,3){_+_}
a: Array[Array[Int]] = Array(Array(0, 1, 2), Array(1, 2, 3))

如何定义函数

def getCol(ith: Int, a: Array[Array[Int]]): Array[Int]

交付

val col2 = getCol(2, a)
col2: Array[Int] = Array(1,2)

一种简单而低效的方法包括

def getCol(ith: Int, a: Array[Int]): Array[Int] = {
  val t = a.transpose
  t(ith)
}

因此也要求更有效的方法。

Answer 1

def getCol(n: Int, a: Array[Array[Int]]) = a.map{_(n - 1)}

请注意，对于第 N 个元素，您必须使用 n - 1。