Skip to main content
 首页 » 编程设计

python之具有移位索引的 Pandas 滚动函数

2025年05月04日20JeffreyZhao

代码

s = pd.Series([0,1,2,3,4]) 
sr = s.rolling(3) 
sr.apply(np.sum) 

返回具有索引 [0,1,2,3,4] 和值 [NaN, NaN, 3, 6, 9] 的系列。是否有快速破解方法,特别是使用 pandas 滚动函数,以便它从以下 3 个索引返回滚动总和,即序列值为 [3, 6 , 9, NaN, NaN]?

请您参考如下方法:

唯一的区别是偏移了 -2:

w = 3 
s.rolling(w).sum().shift(-w + 1) 
 
0    3.0 
1    6.0 
2    9.0 
3    NaN 
4    NaN 
dtype: float64