python - How to find the cumulative sum of numbers in a list? -
time_interval=[4,6,12]
i want sum numbers [4+0, 4+6, 4+6+12]
in order list t=[4,10,22]
. tried:
x=0 in (time_interval): t1=time_interval[0] t2=time_interval[1]+t1 t3=time_interval[2]+t2 print(t1,t2,t3) 4 10 22 4 10 22 4 10 22
if you're doing numerical work arrays this, i'd suggest numpy
, comes cumulative sum function cumsum
:
import numpy np = [4,6,12] np.cumsum(a) #array([4, 10, 22])
numpy faster pure python kind of thing, see in comparison @ashwini's accumu
:
in [136]: timeit list(accumu(range(1000))) 10000 loops, best of 3: 161 per loop in [137]: timeit list(accumu(xrange(1000))) 10000 loops, best of 3: 147 per loop in [138]: timeit np.cumsum(np.arange(1000)) 100000 loops, best of 3: 10.1 per loop
but of course if it's place you'll use numpy, might not worth having dependence on it.
Comments
Post a Comment