2017-09-01 55 views
0

最近,我試圖找到一個有效的方法來做一個系列的累計和。將累加和應用於一系列熊貓.DD

>>> df=pd.DataFrame() 
>>> df['a']=[1,3,1,4,2,5,3,8] 
>>> df 
     a 
    0 1 
    1 3 
    2 1 
    3 4 
    4 2 
    5 5 
    6 3 
    7 8 

預期輸出:

df 
     a b 
    0 1 1 
    1 3 4 
    2 1 5 
    3 4 9 
    4 2 11 
    5 5 16 
    6 3 19 
    7 8 27 

每個b[i]等於sum(a[j] for j<=i)

我通過

df['b']=df.a 
for i in range(df.shape[0]-1): 
    df.b.ix[i+1]+=df.b.ix[i] if df.b.ix[i+1] else df.b.ix[i] 

這是不夠簡潔處理的問題,我想脫掉循環。 在這裏我來諮詢。

回答