2016-12-23 108 views
1

在八度/ Matlab的如下我可以擴大單位矩陣。Scala的微風擴展矩陣

>> 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)?

回答

2

明白了。其實和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(而不是基於在倍頻/ Matlab的1)
包含整數