看板 Python 關於我們 聯絡資訊
※ 引述《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 : ------------------------------------------------------------ : 這個方法小一點的list 可以慢慢推,但我現在有個list 裡面有200000 : 個元素,希望每200個元素作累加,這方法絕對行不通,請教一下有沒有好一點的寫法可 : 以幫忙小弟,謝了 ... length = 200000 block = 200 for i in range(length / block): for j in range(1, block): idx = i * block + j A[idx] += A[idx-1] 一層迴圈不夠 你可以用兩個迴圈XD -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 210.240.114.29 ※ 文章網址: http://www.ptt.cc/bbs/Python/M.1398054454.A.7C7.html ※ 編輯: flarehunter (210.240.114.29), 04/21/2014 12:28:46
Drunk5566:感謝大大!!! 我真想跟你握手!!! 04/21 12:50
COLDTURNIP:from itertools import accumulate 04/21 22:16
COLDTURNIP:B = list(accumulate(a)) # 兩行完工 04/21 22:17
COLDTURNIP:啊抱歉沒看清楚題目,請略過我 XD 04/21 22:19