如何通过索引表在python中索引数组?

时间:2019-07-13 07:27:30

标签: python arrays numpy indexing

我有一个形状为[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]

它们都不起作用。

除了这个问题,我想知道通过这种硬方法进行索引是否比迭代更好?更有效率吗?

0 个答案:

没有答案