python - How to find the cumulative sum of numbers in a list? -
time_interval = [4, 6, 12]
i want sum numbers [4, 4+6, 4+6+12]
in order list t = [4, 10, 22]
.
i tried following:
for 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