arrays - 如何总结Scala数组的每一列？

Question

如果我在 Scala 中有一个数组数组(类似于矩阵)，那么对矩阵的每一列求和的有效方法是什么？例如，如果我的数组数组如下所示:

val arr =  Array(Array(1, 100, ...), Array(2, 200, ...), Array(3, 300, ...))

我想对每一列求和(例如，对所有子数组的第一个元素求和，对所有子数组的第二个元素求和，等等)并得到一个如下所示的新数组:

newArr = Array(6, 600, ...)

我怎样才能在 Spark Scala 中有效地做到这一点？

Answer 1

有合适的 .transpose method在 List这可以帮助这里，虽然我不能说它的效率是什么样的:

arr.toList.transpose.map(_.sum)

(然后调用 .toArray 如果您特别需要将结果作为数组)。