【python】 累積和を取るときのお話

1.型に注意

itertools.accumulate()で返ってくるのはlist型ではないので、要素をindexで指定できません(多分)

from itertools import accumulate
list_1 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

#ng
cumsum_1 = accumulate(list_1)
print("1:",cumsum_1)
print("type:", type(cumsum_1))

print(cumsum_1[1])

結果
1: <itertools.accumulate object at 0x10f8af108>
type: <class 'itertools.accumulate'>
Traceback (most recent call last):
File "test.py", line 9, in <module>
print(cumsum_1[1])
TypeError: 'itertools.accumulate' object is not subscriptable


2.list型にキャストしよう

list型に変更してしまえばこちらのものなので、普通はこうしてしまいます。

#ok
cumsum_2 = list(accumulate(list_1))
print("2:",cumsum_2)
print("type:",type(cumsum_2))

結果
2: [1, 3, 6, 10, 15, 21, 28, 36, 45, 55]
type: <class 'list'>
cumsum_2[1]: 3


3.もっと簡単に(これが書きたかっただけ)

こんな書き方も出来ます
※ ","を忘れると死にます

#ok
* cumsum_3, = accumulate(list_1)
print("3:",cumsum_3)
print("type:",type(cumsum_3))

print("cumsum_3[1]:",cumsum_3[1])

結果
3: [1, 3, 6, 10, 15, 21, 28, 36, 45, 55]
type: <class 'list'>
cumsum_3[1]: 3


-1.余談

for文で回すときに使うだけだったらlistにしなくても問題はないです

for c in accumulate(list_1):
    print(c,end=", ")

1, 3, 6, 10, 15, 21, 28, 36, 45, 55,

あと、なにか補足や豆知識あったらおしえてくだしあ。