改变位置窗口结果

Question

以下是滚动窗口求和的正常结果：

s = pd.Series(range(5))

s.rolling(window=2).sum()

Out[2]: 
0    NaN
1    1.0
2    3.0
3    5.0
4    7.0
dtype: float64

但是，我希望它是

Out[2]: 
0    1.0
1    3.0
2    5.0
3    7.0
4    NaN
dtype: float64

如何通过滚动窗口实现这一点？

提前谢谢你。

Answer 1

让我们用shift来实现吧

window = 2
s.rolling(window = window).sum().shift(-window+1)
Out[270]: 
0    1.0
1    3.0
2    5.0
3    7.0
4    NaN
dtype: float64