將 csv 文件加載到 Breeze DenseMatrix[Double]

Question

我有一個 csv 文件，我想加載到 Breeze DenseMatrix[Double]

這段代碼最終會起作用，但我認為這不是 scala 的處理方式：

  val resource = Source.fromResource("data/houses.txt")
  val lines: Iterator[String] = resource.getLines
  val tmp = lines.toArray
  val numRows: Int = tmp.size
  val numCols: Int = tmp(0).split(",").size
  val m = DenseMatrix.zeros[Double](numRows, numCols)
  //Now do some for loops and fill the matrix

有沒有更優雅和更實用的方法來做到這一點？

Answer 1

  val resource = Source.fromResource("data/houses.txt")
  val lines: Iterator[String] = resource.getLines
  val tmp = lines.map(l => l.split(",").map(str => str.toDouble)).toList
  val m = DenseMatrix(tmp:_*)

好多了