轉換MATLAB風格索引，以numpy的

Question

我想轉換（更復雜的形式）的follwing Matlab代碼

a=zeros(9,9); 
a(3+(1:3),6+(1:3))=1; 

到NumPy的。我試過

from numpy import zeros, r_ 

a=zeros((9,9)) 
a[3+r_[0:3],6+r_[0:3]] = 1 

但是這隻能把3 1放在矩陣中。我如何在Python中以類似（簡短）的形式編寫matlab代碼？

Answer 1

說實話，我很可能只是做

>>> a = zeros((9,9)) 
>>> a[3:6, 6:9] = 1 

或者非硬編碼等效。如果你想索引看起來更像Matlab的在這裏，雖然，那麼你可以使用ix_：

>>> a = zeros((9,9)) 
>>> a[ix_(3+r_[0:3], 6+r_[0:3])] = 1 
>>> a 
array([[ 0., 0., 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 1., 1., 1.], 
     [ 0., 0., 0., 0., 0., 0., 1., 1., 1.], 
     [ 0., 0., 0., 0., 0., 0., 1., 1., 1.], 
     [ 0., 0., 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0., 0., 0.]])