Numpy acumulando una matriz en otra usando la matriz de índice

Question

Mi pregunta es acerca de una operación de matriz específica que quiero expresar usando numpy.

que tienen una serie de flotadores w y una serie de índices idx de la misma longitud que w y quiero resumir todo lo w con el mismo valor idx y recogerlas en una matriz v. como un bucle, esto se parece a esto:

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

¿Hay una manera de hacer esto con las operaciones de la matriz? Mi conjetura era v[idx] += w pero eso no funciona, ya que idx contiene el mismo índice varias veces.

Gracias!

Answer 1

numpy.bincount se introdujo para este propósito:

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

creo que a partir del 1.6 también puede pasar un minLength a bincount.

Answer 2

Este es un comportamiento conocido y, aunque algo desafortunado, no tiene una solución de nivel numpy. (bincount puede usarse para esto si tuerce su brazo.) Hacer el loop usted mismo es realmente su mejor opción.

Tenga en cuenta que el código podría haber sido un poco más clara sin volver a usar el nombre w y sin introducir otra serie de índices, como

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

Si necesita acelerar este bucle, es posible que tenga para bajar a C. Cython lo hace relativamente fácil.