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

发布时间：2020-12-16 09:57:58 所属栏目：安全来源：网络整理

导读：例如,考虑一个二维数组 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(

例如,考虑一个二维数组

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)
}

因此,还要问更有效的方法.

解决方法

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

请注意,您必须使用n – 1作为第N个元素.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!