以与使用元组相同的方式使用ndarray进行索引

2d-array 使用 1d-array 以同样的方式使用元组或基本索引。我的指数是 np.ndarrays 为了方便操作,现在我把它们来回转换成元组。

a = np.zeros((5, 5))
ix = np.array([3, 2])

>>> a[3, 2]
0.0
>>> a[(3, 2)]
0.0
>>> a[ix]
array([[0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0.]])

我试过用很多方法来重塑数组,比如形状 (2, 1) 和 (1, 2) the documentation .