gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-04 08:34:13 26 4
gpt4 key购买 nike

代码

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

关于python - 具有移位索引的 Pandas 滚动函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50419135/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com