numpy.ndenumerate在Fortran命令中返回索引？

Question

當使用numpy.ndenumerate指數返回以下爲C-contiguous排列順序，例如：

import numpy as np 
a = np.array([[11, 12], 
       [21, 22], 
       [31, 32]]) 

for (i,j),v in np.ndenumerate(a): 
    print i, j, v 

沒有馬瑟如果a的order是'F'或'C'，這給：

0 0 11 
0 1 12 
1 0 21 
1 1 22 
2 0 31 
2 1 32 

有任何內置的迭代器numpy像ndenumerate給這個（以下數組order='F'）：

0 0 11 
1 0 21 
2 0 31 
0 1 12 
1 1 22 
2 1 32 

Answer 1

可以按如下方式與np.nditer做到這一點：

it = np.nditer(a, flags=['multi_index'], order='F') 
while not it.finished: 
    print it.multi_index, it[0] 
    it.iternext() 

np.nditer是一個非常強大的野獸，暴露了一些內部Python中的C iterator，請看文檔中的Iterating Over Arrays。

Answer 2

只是走轉置會給你想要的東西：

a = np.array([[11, 12], 
       [21, 22], 
       [31, 32]]) 

for (i,j),v in np.ndenumerate(a.T): 
    print j, i, v 

結果：

0 0 11 
1 0 21 
2 0 31 
0 1 12 
1 1 22 
2 1 32