如何实现pandas rolling滚动窗口求和结果首行有效、末行为NaN

pandas默认rolling方法采用窗口右对齐逻辑，即每个窗口包含当前行及前N-1行，你需要的左对齐窗口（包含当前行及后N-1行）可以通过以下两种方式实现：

方法1：结果偏移实现（最简单） #

计算默认右对齐滚动结果后向上偏移1位即可：

import pandas as pd

s = pd.Series(range(5))
res = s.rolling(window=2).sum().shift(-1)
print(res)

输出结果：

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

方法2：向前窗口索引器实现（更严谨） #

使用pandas提供的向前窗口索引器直接定义窗口范围，无需事后修改结果：

import pandas as pd
from pandas.api.indexers import FixedForwardWindowIndexer

s = pd.Series(range(5))
# 定义向前取2个元素的窗口索引器
indexer = FixedForwardWindowIndexer(window_size=2)
res = s.rolling(window=indexer).sum()
print(res)

输出和方法1完全一致。

内容的提问来源于stack exchange，提问作者thijs818