使用数组索引的numpy数组的2D索引[重复]

Posted 2021-05-06

这个问题在这里已有答案：

• Indexing of 3d numpy arrays with 2d arrays 1回答

如果我有一个2D数组的索引：

i = np.array([[0, 0], [1, 1]])

还有一个我要索引的2D数组：

a = np.array([[1, 2], [3, 4]])

如何索引数组以获得如下所示的一维数组？

np.array([1, 4])

答案

您可以使用：

a[i.T[0], i.T[1]]

如果您有更多尺寸，您可以使用：

a[tuple(i.T)]

另一答案

你只需要一个简单的索引：

In [11]: a[i[:, 0], i[:, 1]]
Out[11]: array([1, 4])

i的第一列是行数，第二列表示列号。