从scala中的双数组列表创建Breeze DenseMatrix

时间:2018-01-09 10:53:55

标签: scala scala-breeze

我有一个结构List[Array[Double]],我想转换为DenseMatrix。我有这个解决方案,但我认为可能有更好的方法:

val data = List[Array[Double]]
val rows = data.length;
val cols = data(0).length;
val matrix = DenseMatrix.zeros[Double](rows, cols)
for (i <- 0 until data.length) {
  for (j <- 0 until data(i).length) {
    matrix(i,j)= data(i)(j)

  }
}

我一直在浏览Breeze文档,但没有找到任何内容。还有更好的方法吗?

1 个答案:

答案 0 :(得分:3)

你可以试试这个:

val matrix = DenseMatrix(data:_*)

<强> EDIT1

有关其工作原理的解释,您可以将data: _*视为扩展为变量参数。例如,如果

val data = List[Array[Double]](Array(1.0, 1.0), Array(2.0, 2.0))

然后DenseMatrix(data:_*)DenseMatrix(Array(1.0, 1.0), Array(2.0, 2.0))

相同

更多详情: