2017-02-28 94 views
0

我想累積基於ID和月份的數據使用量,以便將累積數據作爲附加列添加。根據df中的兩列值累積數據python

這是我對DF的樣本代碼:

df = pd.DataFrame({'ID':["SAD1", "SAD2", "SAD1", "SAD2", "SAD1", "SAD2", "SAD3"], 
       'Month':["201701","201701","201702","201702","201703","201703", "201703"], 
       'Usage':[20,40,100,50, 30, 30, 2000]}) 

我的數據框看起來是這樣的:

 ID Month Usage 
0 SAD1 201701  20 
1 SAD2 201701  40 
2 SAD1 201702 100 
3 SAD2 201702  50 
4 SAD1 201703  30 
5 SAD2 201703  30 
6 SAD3 201703 2000 

我想結束了,結果是:

 ID Month Usage Cum Usage 
0 SAD1 201701  20   20 
1 SAD2 201701  40   40 
2 SAD1 201702 100  120 
3 SAD2 201702  50   90 
4 SAD1 201703  30  150 
5 SAD2 201703  30  120 
6 SAD3 201703 2000  2000 

任何人都可以幫助我如何將這個應用到我的DF?

+0

'DF [ '射精用法'] = df.groupby([ 'ID'])。cumsum()'? – Abdou

回答

4

groupbycumsum

df['Cum Usage'] = df.groupby('ID').cumsum()