Numpy che accumula un array in un altro utilizzando l'array di indici

Question

La mia domanda riguarda un'operazione di array specifica che desidero esprimere utilizzando numpy.

ho un array di carri w e una serie di indici idx della stessa lunghezza w e voglio riassumere tutti w con lo stesso valore idx e raccoglierli in un array v. come un ciclo, questo si presenta così:

for i, x in enumerate(w): 
    v[idx[i]] += x 

C'è un modo per fare questo con le operazioni di matrice? La mia ipotesi era v[idx] += w ma questo non funziona, dal momento che idx contiene lo stesso indice più volte.

Grazie!

Answer 1

numpy.bincount è stato introdotto per questo scopo:

tmp = np.bincount(idx, w) 
v[:len(tmp)] += tmp 

penso al 1.6 si può anche passare un minlength a bincount.

Answer 2

Questo è un comportamento noto e, sebbene in qualche modo sfortunato, non presenta una soluzione alternativa a livello di disturbo. (bincount può essere usato per questo se si torce il suo braccio.) Fare il ciclo te stesso è davvero la soluzione migliore.

Nota che il codice avrebbe potuto essere un po 'più chiaro, senza ri-utilizzando il nome w e senza introdurre un'altra serie di indici, come

for i, w_thing in zip(idx, w): 
    v[i] += w_thing 

Se avete bisogno di accelerare questo ciclo, si potrebbe avere passare a C. Cython rende questo relativamente facile.