与pandas中的df.diff()相反

时间:2018-01-24 17:47:09

标签: python pandas

我在论坛中搜索了一种更简洁的方法,在数据框中创建一个新列,该行是前一行的总和 - 与.diff()函数相反,后者取差。

这就是我目前正在解决问题的方法

df = pd.DataFrame ({'c':['dd','ee','ff', 'gg', 'hh'], 'd':[1,2,3,4,5]}
df['e']= df['d'].shift(-1)
df['f'] = df['d'] + df['e']

您的意见表示赞赏。

2 个答案:

答案 0 :(得分:5)

您可以使用窗口大小为2且rolling的{​​{1}}:

sum

答案 1 :(得分:2)

df.cumsum()

Example:
data = {'a':[1,6,3,9,5], 'b':[13,1,2,5,23]}
df = pd.DataFrame(data)

df = 
    a   b
0   1   13
1   6   1
2   3   2
3   9   5
4   5   23

df.diff()

a   b
0   NaN NaN
1   5.0 -12.0
2   -3.0    1.0
3   6.0 3.0
4   -4.0    18.0

df.cumsum()

a   b
0   1   13
1   7   14
2   10  16
3   19  21
4   24  44