如果我有一个二维数组的索引：
i = np.array([[0, 0], [1, 1]])
还有一个我想索引的二维数组：
a = np.array([[1, 2], [3, 4]])
我如何索引数组以获得一维数组，如下所示？
np.array([1, 4])
您可以使用：
a[i.T[0], i.T[1]]
如果您有更多尺寸，可以使用：
a[tuple(i.T)]
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
点击生成二维码
我来说两句