python - 用另一个数组索引 Numpy 数组的有效方法

Question

给定一个 Numpy 数组x和一个整数数组y，我想做一些相当于：

z = np.array(x[i] for i in y)

是否有一个 Numpy 函数/方法可以有效地执行此操作而无需转换回列表？

Answer 1

如果y包含对 有效的索引x，则：

z = x[y]

>>> import numpy as np
>>> x = np.arange(100)
>>> y = np.array([1, 27, 36, 98])
>>> x[y]
array([ 1, 27, 36, 98])