我正在尝试在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})
显然错了,这样做的理想方法是什么?
由于
答案 0 :(得分:2)
使用transpose
将x
的列转换为行,这样可以更轻松 - 然后可以使用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