scala - 在 Scala 中对数组数组执行元素添加的惯用方法是什么

Question

我有一个 Array[Array[Float]]保证所有内部数组的长度相同。我想对所有内部数组执行元素加法。

让我用一个具体的例子来说明这一点。假设我的数组包含三个数组，由七个浮点数组成，具有以下值:

Array[Array[Float]] = Array(
                          Array(0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0), 
                          Array(1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0), 
                          Array(0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0)
                      )

现在我想把它转换成一个大小为 7 的数组，其中第 n 个元素是我输入中第 n 个元素的总和 Array .对于我的示例输入，输出将是:

Array[Float] = Array(1.0, 3.0, 3.0, 5.0, 5.0, 7.0, 7.0)

用于在两个 Array[Float] 上执行此操作我知道我可以压缩它们，然后用 map 添加元素。然而，我仍然很难找到一个更通用的解决方案，它能够对任意数量的 Array[Float] 求和。的元素。

任何帮助将非常感激!

Answer 1

scala> a.transpose.map(_.sum)
res1: Array[Double] = Array(1.0, 3.0, 3.0, 5.0, 5.0, 7.0, 7.0)