多个维度的花式索引

假设

import numpy as npnp.random.seed(0)n,m,k = 2,3,5A = np.arange(n*m*k,0,-1).reshape((n,m,k))print(A)# [[[30 29 28 27 26]#   [25 24 23 22 21]#   [20 19 18 17 16]]#  [[15 14 13 12 11]#   [10  9  8  7  6]#   [ 5  4  3  2  1]]]B = np.random.randint(k, size=(n,m))print(B)# [[4 0 3]#  [3 3 1]]

要创建此数组，

print(A.reshape(-1, k)[np.arange(n * m), B.ravel()])# [26 25 17 12  7  4]

作为

nxm

i,j = np.ogrid[0:n, 0:m]print(A[i, j, B])# [[26 25 17]#  [12  7  4]]