Scala:在一个结构中合并两个数组

时间:2017-01-19 15:45:06

标签: arrays scala

我有两个数组:

val diceProbDist = new Array[Double](2 * DICE + 1)

val diceExpDist = new Array[Double](2 * DICE + 1)

我希望合并到一个单独的结构中(可能是某种元组):

(0, 0.0, 0,0)(1, 0.0, 0.0)(2, 0.02778, 0.02878)...

其中第一个条目是数组索引,第二个条目是第一个数组值,第三个条目是第二个数组值。

是否有一些scala函数可以实现(使用地图或类似的拉链)?

感谢, ML

3 个答案:

答案 0 :(得分:7)

val diceProbDist = Array(0.1, 0.2, 0.3)
val diceExpDist = Array(0.11, 0.22, 0.33)

diceProbDist
 .zip(diceExpDist)
 .zipWithIndex
 .map { case ((v1, v2), i) => (i, v1, v2) }

// result: Array((0,0.1,0.11), (1,0.2,0.22), (2,0.3,0.33))

答案 1 :(得分:4)

如果您不介意,简单的for理解也应该可以解决问题:

for {
  index <- 0 until math.min(diceExpDist.length, diceProbDist.length)
} yield (index, diceProbDist(index), diceExpDist(index))

答案 2 :(得分:2)

另一个解决方案,类似于tkachuko的一个没有理解

的解决方案
val diceProbDist = List(0.1, 0.2, 0.3)
val diceExpDist = List(0.11, 0.22, 0.33)

val range = 0 until math.min(diceExpDist.length, diceProbDist.length)
range.map { idx => (idx, c(idx), d(idx)) }

// result : res0: List[(Int, Int, Int)] = List((0,0,11), (1,1,12), (2,2,13), (3,3,14), (4,4,15), (5,5,16), (6,6,17), (7,7,18), (8,8,19), (9,9,20))