如果我有一个2D数组索引：
i = np.array([[0, 0], [1, 1]])
我要索引的2D数组：
a = np.array([[1, 2], [3, 4]])
如何索引数组以获得如下所示的一维数组？
np.array([1, 4])
答案 0 :(得分：3)
您可以使用：
a[i.T[0], i.T[1]]
如果您有更多尺寸，可以使用：
a[tuple(i.T)]
答案 1 :(得分：0)
您只需要一个简单的索引：
In [11]: a[i[:, 0], i[:, 1]]
Out[11]: array([1, 4])
i的第一列是行数，第二列表示列号。