Pandas滚动窗口，包括前面的“n”行和后面的“m”行

Question

在pandas中，如何编写一个既有前面行又有后面行的rolling window

例如，sql滚动窗口可以按如下方式编写：

sum(id_0) over (partition by id_1 order by id_2 rows between 3 preceding and 1 following)

我不知道如何用熊猫来写这个

Answer 1

您可以计算前四个值的滚动和，然后使用shift()方法将这些值应用于前一行-例如：

rolling_ds = dataset.rolling(4, min_periods=1).sum().shift(-1)

关于这种方法的更有力的讨论可以在这里找到：Calculating Rolling forward averages with pandas