带有Lamdba的熊猫滚动功能

Question

我正在开发带有代码的旧熊猫0.24.0版本：

df["A"] = df['B'].rolling(window=2).apply(lambda x: x[0] - x[1])

但是，熊猫版本已升级到1.1.0，导致此代码无法正常工作。 我也尝试过使用滚动功能的其他变体以及扩展功能。

有人可以帮忙编写以上代码以在 pandas 1.1.0 中工作吗？

谢谢

Answer 1

这更像shift

df['A'] = df['B'] - df['B'].shift()