scala - 如何根据某些条件替换 Scala 中微风矩阵的元素？

Question

我正在 Scala 中使用二维 Breeze 矩阵。在某些时候，我必须对两个矩阵进行逐元素除法。分母矩阵中的某些元素可能为零，从而导致结果中出现 NaN。

我可以遍历矩阵维度并将 0.0 替换为 > 0 的值。

但是有没有更简单的或 Scala 惯用的解决方案呢？

Answer 1

一步步:

使用示例矩阵:

val dm = DenseMatrix((1.0, 0.0, 3.0), (0.0, 5.0, 6.0))

找出哪些元素等于 0.0:

dm :== 0.0

breeze.linalg.DenseMatrix[Boolean] =
false  true   false
true   false  false

切片矩阵:

dm(dm :== 0.0)

breeze.linalg.SliceVector[(Int, Int),Double] = breeze.linalg.SliceVector@2b

使用切片矩阵进行替换:

dm(dm :== 0.0) := 42.0

breeze.linalg.Vector[Double] = breeze.linalg.SliceVector@2b

检查矩阵:

dm

breeze.linalg.DenseMatrix[Double] =
1.0   42.0  3.0
42.0  5.0   6.0