按矩阵切片Numpy数组

Question

考虑形状为（m，n，k）的Numpy数组 A 。 进一步让 B 为形状为（m，n）的Numpy数组，并在 {0,1，...，k-1} 。 我现在想知道一种有效的方法（避免嵌套循环）来定义由

给出的形状为（m，n）的Numpy数组 C

C[i,j] = A[i,j,B[i,j]]

对于0 <= i