如何在熊猫数据框上执行滑动窗口操作？

Question

这与my previous question相同，但是这次窗口可以重叠。

我想控制窗口大小，[以及边缘会发生什么（尽管现在“相同”的填充将起作用）]。

输入：

a
0
5
2
1
8
2
5

window_size=3，op=np.max

结果：

a res
0 5
5 5
2 5
1 8
8 8
2 8
5 5

---

我还希望传递通用函数，以减少列数和减少行数，如链接的问题所示。

---

奖金：

如果我不需要计算所有值，我可以做一个步幅> 1的步幅运行窗口，并在其余部分填写nans。可以这样做来保存计算吗？

Answer 1

您想要这样的东西

df['res'] = df['a'].rolling(3, min_periods=0, center=True).max()

对于通用函数，应将.max()替换为.aggregate(fn)