Scala Breeze 扩展矩阵

Scala Breeze expansion matrix

在 Octave/Matlab 中,我可以如下展开单位矩阵。

>> I = eye(3)
I =
Diagonal Matrix
  1   0   0
  0   1   0
  0   0   1

>> A = [ 3 2 3 2 2 1 3 2 2 1 ]

>> E = I(:, A)
E =
   0   0   0   0   0   1   0   0   0   1
   0   1   0   1   1   0   0   1   1   0
   1   0   1   0   0   0   1   0   0   0

如何使用 Scala/Breeze 实现同样的事情(即从 A 获取 E,可能使用 I)?

知道了。实际上与Octave非常相似。

scala> val I = DenseMatrix.eye[Int](3)
I: breeze.linalg.DenseMatrix[Int] =
1  0  0
0  1  0
0  0  1

scala> val A = DenseMatrix(2, 1, 2, 1, 1, 0, 2, 1, 1, 0) // zero based in breeze

scala> I(::, A.toArray.toSeq)
res26: breeze.linalg.SliceMatrix[Int,Int,Int] =
0  0  0  0  0  1  0  0  0  1
0  1  0  1  1  0  0  1  1  0
1  0  1  0  0  0  1  0  0  0

注意事项是:

  • 矩阵必须包含整数
  • 索引基于 0(相对于 Octave/Matlab 中的 1)