2011-04-27 24 views
12

que tiene una matriz numpy comoCorrer suma acumulada de 1d matriz NumPy

x=np.array([1,2,3,4])

quiero crear otra matriz numpy y que es la suma acumulada de x, de modo que

y=np.array([1,3,6,10])

¿Cuál es una buena forma num-pythonic para hacer esto?

Respuesta

7

Otra opción es:

y = np.add.accumulate(x) 

que es muchas veces más rápido que np.cumsum pesar de que la documentación dice que son equivalentes.

+1

¿Alguna idea de por qué es más rápido? – endolith