Scala - 在已经声明的数组中初始化值的最优雅方式？

Question

我有一个像这样定义的3d数组:

val 3dArray = new Array[Array[Array[Int]]](512, 8, 8)

在Javascript中,我将执行以下操作将每个元素分配给1:

for (i = 0; i < 512; i++)
 {
    3dArray[i] = [];
    for (j = 0; j < 8; j++)
    {
        3dArray[i][j] = [];
        for (k = 0; k < 8; k++)
        {
            3dArray[i][j][k] = 1;
        }
    }
}

做同样的事情最优雅的方法是什么？

Answer 1

不确定有一个特别优雅的方法,但这是一种方式(我使用后缀s来表示维度,即xss是一个二维数组).

val xsss = Array.ofDim[Int](512, 8, 8)
for (xss <- xsss; xs <- xss; i <- 0 until 8) 
  xs(i) = 1

或者,使用变换它甚至更短:

for (xss <- xsss; xs <- xss)
  xs transform (_ => 1)

Answer 2

for {
    i <- a.indices
    j <- a(i).indices
    k <- a(i)(j).indices
} a(i)(j)(k) = 1

要么

for {
    e  <- a
    ee <- e
    i  <- ee.indices
} ee(i) = 1