Scala:如何使用数组的操作

时间:2016-05-01 12:47:44

标签: arrays scala operation

我正在尝试在Scala中对数组进行操作。

x是2d阵列,y是1d阵列。例如

val x = Array(Array(1.0,2.0,3.0),Array(4.0,5.0,6.0),Array(7.0,8.0,9.0),Array(10.0,11.0,12.0).....Array(..))
val y = Array(2.0,3.0,4.0,5.0,.....n)

假设x.length = y.length

我希望获得Array[Double](3)之类的

Array(1.0*2.0 + 4.0*3.0 + 7.0 * 4.0 + 10.0 * 5.0 ...., 2.0*2.0 + 5.0*3.0 + 8.0*4.0 + 11.0 * 5.0 ....., 3.0*2.0 + 6.0 * 3.0 + 9.0 * 4.0 + 12.0 * 5.0....)

我不知道该怎么做。

我使用了x.map(x => (x zip y).map{case(a, b) => a * b})

显然错了,这样做的理想方法是什么?

由于

1 个答案:

答案 0 :(得分:2)

使用transposex的列转换为行,这样可以更轻松 - 然后可以使用y压缩每列,然后可以将值相乘总结:

val x = Array(Array(1.0,2.0,3.0),Array(4.0,5.0,6.0),Array(7.0,8.0,9.0),Array(10.0,11.0,12.0))
val y = Array(2.0,3.0,4.0,5.0)

val result = x.transpose.map(_.zip(y).map { case (a, b) => a*b }.sum )

print(result.mkString(",")) // prints 92.0,106.0,120.0