我有两个数组:
一个我想到的解决方案是这样做:
A
,使得A.shape = (N, M, K)
,和index
,使得index.shape = (N, M)
。因此,我想得到这样的数组B
,即B[i, j] = A[i, j, index[i, j]]
。如何以最简单的方式实现它?一个我想到的解决方案是这样做:
B = A.reshape(-1, K)[np.arange(N * M), index.reshape(-1)].reshape(N, M)
。我想知道是否还有其他解决方案?