如果我在 Scala 中有一个数组的数组(类似于矩阵),那么对矩阵的每一列求和的有效方法是什么?例如,如果我的数组数组如下所示:
val arr = Array(Array(1, 100, ...), Array(2, 200, ...), Array(3, 300, ...))
我想对每一列求和(例如,对所有子数组的第一个元素求和,对所有子数组的第二个元素求和等)并得到一个新数组,如下所示:
newArr = Array(6, 600, ...)
如何在 Spark Scala 中高效地完成此操作?
有一个合适的.transpose method on List
尽管我不能说它的效率如何,但它可以在这里提供帮助:
arr.toList.transpose.map(_.sum)
(然后调用.toArray
如果您特别需要结果作为数组)。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)