用索引矩阵检索值的Python方法？

Question

我有一个h-by-w-by-3 numpy数组A和另一个h-by-w-by-2 numpy数组B，它告诉我从哪里获取值A在每个（h，w）位置。用C（h-by-w-by-3）表示结果矩阵。

天真，我可以做

C = np.zeros(A.shape)
for i in range(B.shape[0]):
    for j in range(B.shape[1]):
        C[i, j, :] = A[B[i, j, 0], B[i, j, 1], :]

但是我敢肯定，有一种更高效，更pythonic和更快的方式！

Answer 1

确实有：

runBlocking()