Numpy: riorganizzare l'array in base all'array dell'indice

Question

Sto cercando una soluzione a una riga che possa aiutarmi a fare quanto segue.

Supponiamo che io sono

array = np.array([10, 20, 30, 40, 50]) 

mi piacerebbe riorganizzare esso sulla base di un ordine di ingresso. Se ci fosse una funzione NumPy chiamato arrange, sarebbe effettuare le seguenti operazioni:

newarray = np.arrange(array, [1, 0, 3, 4, 2]) 
print newarray 

    [20, 10, 40, 50, 30] 

Formalmente, se la matrice da riordinare è mxn, e il "index" array è 1 xn, l'ordine sarebbe determinata dalla array chiamato "index".

Il numpy ha una funzione come questa?

Answer 1

Si può semplicemente utilizzare il "indice" lista direttamente, come, pure, una matrice di indice:

>>> arr = np.array([10, 20, 30, 40, 50]) 
>>> idx = [1, 0, 3, 4, 2] 
>>> arr[idx] 
array([20, 10, 40, 50, 30]) 

tende ad essere molto più velocemente se idx è già un ndarray e non un list, anche se Funzionerà in entrambi i modi:

>>> %timeit arr[idx] 
100000 loops, best of 3: 2.11 µs per loop 
>>> ai = np.array(idx) 
>>> %timeit arr[ai] 
1000000 loops, best of 3: 296 ns per loop