Numpyでスライスからエントリを選択

Question

形状A（3,10,10）、形状B（10,10）の配列Bを仮定します。 Bの各要素は、Aの第1の軸に沿った要素のインデックスです。？

Answer 1

使用高度なインデックスと放送：

import numpy as np 

L, M, N = 3, 8, 10 
A = np.arange(L)[:, None, None] + 10*np.arange(M)[:, None] + 100*np.arange(N) 
B = np.random.randint(0, L, (M, N)) 
m, n = np.ogrid[:M, :N] 
C = A[B, m, n] 
np.all(C%10 == B) 
# True