在Scala中制作归零数组的更快方法

时间:2014-05-13 14:09:27

标签: arrays scala

我在Scala中创建了归零阵列 (0 until Nrows).map (_ => 0).toArray但有什么更快的吗? map很慢。

我有同样的问题,但用1而不是O,即我也想加速(0 until Nrows).map (_ => 1).toArray

2 个答案:

答案 0 :(得分:9)

零是Ints数组的默认值,所以只需执行以下操作:

    val array = new Array[Int](NRows)

如果你想要所有这些值都是1,那么使用.fill()(感谢@gourlaysama):

    val array = Array.fill(NRows)(1)

然而,看看它如何在内部工作,它涉及创建一些你不需要的对象。如果速度是您的主要考虑因素,我怀疑以下(丑陋)方法可能会更快:

    val array = new Array[Int](NRows)
    for (i <- 0 until array.length) { array(i) = 1 }

答案 1 :(得分:6)

对于多维数组,请考虑Array.ofDim,例如

scala> val a = Array.ofDim[Int](3,3)
a: Array[Array[Int]] = Array(Array(0, 0, 0), Array(0, 0, 0), Array(0, 0, 0))

同样地,

scala> val a = Array.ofDim[Int](3)
a: Array[Int] = Array(0, 0, 0)

在此处的上下文中,

val a = Array.ofDim[Int](NRows)

要设置(可能非零)初始值,请考虑Array.tabulate,例如

scala> Array.tabulate(3,3)( (x,y) => 1)
res5: Array[Array[Int]] = Array(Array(1, 1, 1), Array(1, 1, 1), Array(1, 1, 1))

scala> Array.tabulate(3)( x => 1)
res18: Array[Int] = Array(1, 1, 1)