Матрица расширения Scala Breeze

Question

В 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 

Как можно достичь того же (то есть получить Е из А, возможно, с использованием I) с Scala/Бриз?

Answer 1

Got it. На самом деле очень похож на Октав.

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 на основе (в отличие от 1, базирующейся в Octave/Matlab)