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])