Re: [問題] list 裡的元素累加

作者: flarehunter (Range)   2014-04-21 12:27:31
※ 引述《Drunk5566 (啾咪)》之銘言:
: 不好意思,小弟有一個小問題想發問,題目如下:
: 我有一個 list , A = [1,2,3,4,5,6,7,8,9,10] , 想要將裡面的元素作累加,並且加到
: 第五個元素停止,第六個元素在重新作累加,也就是希望作出一個新的list B, 裡面的元
: 素為 B = [1 , 1+2 , 1+2+3 , 1+2+3+4 , 1+2+3+4+5, 6, 6+7, 6+7+8, 6+7+8+9,
: 6+7+8+9+10]
: 我寫得如下:
: if i<5:
: x = x + A[i]
: print x
: else:
: y = y + A[i]
: print y
: 1
: 3
: 6
: 10
: 15
: 6
: 13
: 21
: 30
: 40
:
作者: Drunk5566 (啾咪)   2014-04-21 12:50:00
感謝大大!!! 我真想跟你握手!!!
作者: COLDTURNIP (娶妻當娶五航戰)   2014-04-21 22:16:00
from itertools import accumulateB = list(accumulate(a)) # 兩行完工啊抱歉沒看清楚題目,請略過我 XD

Links booklink

Contact Us: admin [ a t ] ucptt.com