Scala中的上三角矩阵
发布时间:2020-12-16 10:05:27 所属栏目:安全 来源:网络整理
导读:有没有办法可以在 scala中更快地计算上三角矩阵? /** Returns a vector which consists of the upper triangular elements of a matrix */ def getUpperTriangle(A: Array[Array[Double]]) = { var A_ = Seq(0.) for (i - 0 to A.size - 1;j - 0 to A(0).si
有没有办法可以在
scala中更快地计算上三角矩阵?
/** Returns a vector which consists of the upper triangular elements of a matrix */ def getUpperTriangle(A: Array[Array[Double]]) = { var A_ = Seq(0.) for (i <- 0 to A.size - 1;j <- 0 to A(0).size - 1) { if (i <= j){ A_ = A_ ++ Seq(A(i)(j)) } } A_.tail.toArray } 解决方法
我不知道更快,但这更短,更“功能”(我注意到你用功能编程标记了你的问题)
def getUpperTriangle(a: Array[Array[Double]]) = (0 until a.size).flatMap(i => a(i).drop(i)).toArray 或者,或多或少相同的想法: def getUpperTriangle(a: Array[Array[Double]]) = a.zipWithIndex.flatMap{case(r,i) => r.drop(i)} (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |