[python] accumulate(itertools), 누적 합

📗 accumulate

뜻으로는 축적하다?

한마디로 누적된 합을 뽑아주는 녀석

🔵 사용법

from itertools import accumulate

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
b = list(accumulate(a))
print(a)  # [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
print(b)  # [1, 3, 6, 10, 15, 21, 28, 36, 45, 55]

값이 저절로 누적되는 것을 볼 수 있다.

이것을 왜 사용하느냐하면

사실 for문 만으로도 값을 누적한 리스트를 뽑을 수 있지만 속도면에서 큰 차이가 난다.( 특히 값이 커지면 커질수록 )

 

🔵 속도 측정

1번 for문

a = [x+1 for x in range(1000000)]
for i in range(1, len(a)):
    a[i] += a[i-1]

2번 accumulate

a = [x+1 for x in range(1000000)]
b = list(accumulate(a))

실행 코드	값	속도
for문	10만	0.01894974708557129
accumulate	10만	0.004986286163330078
for문	1000만	1.868004560470581
accumulate	1000만	0.766948938369751

accumulate가 확실히 빠른것을 볼 수 있다. 많이 사용할 일은 없겠지만 알아두면 좋을 것 같다.