计算跨特定跨度的pandas数据帧中的累积总和

时间:2017-06-24 15:51:04

标签: python pandas cumulative-sum

我有这个人。数据帧:

            tot_pix  Season  caap_col    lamma     kite
datetime                                                  
2000-01-01   1914.0  2000.0       1.0  1.95025  117.737362
2000-01-04   1914.0  2000.0       1.0  1.95025  117.674177
2000-01-05   1914.0  2000.0       1.0  1.95025  117.995489
2001-01-04   1914.0  2001.0       1.0  1.95025  118.114809
2001-01-05   1914.0  2001.0       1.0  1.95025  118.160295

在上面的数据框中,我想计算kite列的累积总和。但是,我希望累积总和不跨越Season值。例如。 kite的累计和的输出应如下所示:

          tot_pix Season caap_col lamma kite
datetime                    
1/1/2000    1914    2000    1   1.95025 117.737362
1/4/2000    1914    2000    1   1.95025 235.411539
1/5/2000    1914    2000    1   1.95025 235.669666
1/4/2001    1914    2001    1   1.95025 118.114809
1/5/2001    1914    2001    1   1.95025 236.275104

我可以使用cumsum命令计算累积总和,如何将其限制为Season的特定范围?

1 个答案:

答案 0 :(得分:3)

您需要groupby + cumsum

DrawLine