如何通过 2 个数组中给出的索引引用 2D 数组中的元素?

Question

假设我有一个二维数字数组：

a = np.array([[1,2,3],
[4,5,6],
[7,8,9]])

和 2 个用于索引的数组：

ind1 = np.array([0,1,1],
[0,2,1],
[1,2,1])
ind2 = np.array([[2,2,1],
[0,1,2],
[1,0,0]])

我想创建一个与a形状相同的新数组b，以便b[i,j] = a[ind1[i,j],ind2[i,j]].如何有效地做到这一点？

Answer 1

你可以简单地做

b = a[ind1, ind2]

输出

[[3 6 5]
[1 8 6]
[5 7 4]]