首页 > 解决方案 > 大熊猫中带有移位索引的滚动总和

问题描述

给定以下数据框

import pandas as pd
import numpy as np

df = pd.DataFrame({'col1': [1, 2, 3, 4, 5, 6, 7, -9]})
df = df['col1'].rolling(3).mean()

但是,我想要一个移动的滚动范围,它给我每行的平均值,索引 i 作为索引 i-4、i-5 和 i-6 的平均值。

x_i = (x_{i-4} + x_{i-5} + x_{i-6})/3

有没有办法通过移位和滚动来做到这一点?

所需输出 NAN, NAN, NAN, NAN, NAN, NAN, (1+2+3)/3, (2+3+4)/3

标签: pandas

解决方案


推荐阅读