Pandas cumsum具有初始值

时间:2015-10-03 11:11:31

标签: python pandas cumsum

所以给出以下单列数据框:

from pandas import DataFrame
df = DataFrame({0: {'10/10/2012': 50, '10/11/2012': -10, '10/12/2012': 100})

In [3]: df
Out[3]: 
              0   
10/10/2012   50   
10/11/2012  -10  
10/12/2012  100  

我想为给定列创建累积和,但为累加器提供初始值。因此对于上面的数据帧并且初始值为100,这看起来像:

              0  1 
10/10/2012   50  150 
10/11/2012  -10  140
10/12/2012  100  240

任何帮助,非常感谢。

1 个答案:

答案 0 :(得分:2)

from pandas import DataFrame

df = DataFrame({0: {'10/10/2012': 50, '10/11/2012': -10, '10/12/2012': 100}})

print(df)

这会给你以下结果

              0
10/10/2012   50
10/11/2012  -10
10/12/2012  100

df[1] = df[0].cumsum()+100

print(df)

创建新列df[1]后,获取df[0]的累积总和,然后将其加100。

              0    1
10/10/2012   50  150
10/11/2012  -10  140
10/12/2012  100  240