python - 与 pandas 中的 df.diff() 相反

Question

我搜索了论坛以寻找一种更简洁的方法来在数据框中创建一个新列，该列是该行与前一行的总和 - 与取差值的 .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']

感谢您的想法。

Answer 1

您可以使用窗口大小为 2 的 rolling 和 sum:

df['f'] = df['d'].rolling(2).sum().shift(-1)

    c  d    f
0  dd  1  3.0
1  ee  2  5.0
2  ff  3  7.0
3  gg  4  9.0
4  hh  5  NaN