如何对pandas数据框进行滑动窗口操作？

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 进行跨步运行窗口，并在其余部分填写 nan。这样做可以节省计算吗？

Answer 1

你想要这样的东西df['res'] = df['a'].rolling(3, min_periods=0, center=True).max()对于通用函数，您应该替换.max()为.aggregate(fn)