我有一个形状为[N,M,H]的数组A 现在我要为每行N和每列M编制索引,第三个维度由索引表给出。即:
A[0,0,table[0,0]],A[0,1,table[0,1]],..........,A[0,M-1,table[0,M-1]]
...
...
A[N-1,0,table[N-1,0]],A[N-1,1,table[N-1,0]],...,A[N-1,M-1,table[N-1,M-1]]
One (N,M) matrix indexed by the table,that table range from 0 to H-1
我尝试了两种方法:
1.A[:,:,table]
2.A[range(N),range(M),table]
它们都不起作用。
除了这个问题,我想知道通过这种硬方法进行索引是否比迭代更好?更有效率吗?