moyen efficace pour indexer un tableau Numpy avec un autre tableau

Question

Étant donné un tableau numpy x, et un tableau d'entiers y, je veux faire quelque chose d'équivalent à:

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

Y at-il une fonction Numpy/méthode faire cela efficacement sans avoir à revenir à une liste?

Answer 1

Si y contient des indices qui sont valables pour x, puis:

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