我有一个 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 和更快的方式!
跃然一笑
相关分类