如何以更优雅的Scala函数式方式加载CSV到Breeze DenseMatrix[Double]

你原来的代码需要手动初始化矩阵再循环填充，确实不够贴合Scala的函数式编程风格。可以利用Scala集合的链式操作+Breeze内置的构造器，实现更简洁的方案：

基础实现 #

import scala.io.Source
import breeze.linalg.DenseMatrix

// 读取文件并直接转换为DenseMatrix
val denseMatrix = Source.fromResource("data/houses.txt").getLines()
  // 每行分割为字符串数组后转成Double数组
  .map(_.split(",").map(_.toDouble))
  .toArray
  // 利用Scala可变参数展开语法，直接传入二维数组构造矩阵
  .pipe(DenseMatrix(_: _*))

带合法性检查的健壮版本 #

如果需要处理空文件、行长度不一致等异常情况，可以添加前置检查：

import scala.io.Source
import breeze.linalg.DenseMatrix

val rowIter = Source.fromResource("data/houses.txt").getLines()
  .map(line => line.split(",").map(_.toDouble))
  .toList

// 检查文件非空
require(rowIter.nonEmpty, "CSV文件不能是空文件")
val columnCount = rowIter.head.length
// 检查所有行的列数一致
require(rowIter.forall(_.length == columnCount), "CSV文件中所有行的列数必须相同")

val denseMatrix = DenseMatrix(rowIter.toArray: _*)

代码说明 #

• 全程使用函数式的链式调用，避免了手动初始化零矩阵和循环填充的冗余代码
• 利用Breeze的DenseMatrix构造器支持直接接收二维数组的特性，结合Scala的_*语法展开数组为可变参数
• 健壮版本通过require提前校验数据合法性，避免后续构造矩阵时出现不可预期的错误

内容的提问来源于stack exchange，提问作者MLeiria