熊猫滚动意味着更新

考虑数据框：

df = pd.DataFrame({
    "a": [None, None, None, None, 1, 2, -1, 0, 1],
    "b": [5, 4, 6, 7, None, None, None, None, None]
})

>>  a    b
0   NaN  5.0
1   NaN  4.0
2   NaN  6.0
3   NaN  7.0
4   1.0  NaN
5   2.0  NaN
6  -1.0  NaN
7   0.0  NaN
8   1.0  NaN

对于每个缺失值b我想取前 4 个值的平均值加上中的值a具有相同的索引。例如，7点之后：

4: (5   + 4 + 6 + 7) / 4 + 1 = 6.5
5: (6.5 + 4 + 6 + 7) / 4 + 2 = 7.88
   ...

结果数据框应该是：

>>  a    b
0   NaN  5.00
1   NaN  4.00
2   NaN  6.00
3   NaN  7.00
4   1.0  6.50
5   2.0  7.88
6  -1.0  5.84
7   0.0  6.80
8   1.0  7.76

如何实现这一目标？

在这里使用 for 循环，panda 不是 row-wise 的，它们不能使用先前计算的值进行将来的计算。（矢量化）

l=[]
for x ,y in zip(*df.values.T.tolist()):
    if len(l)<4:
        l.append(y)
    else:
        l.append(sum(l[-4:])/4+x)

l
Out[188]: [5.0, 4.0, 6.0, 7.0, 6.5, 7.875, 5.84375, 6.8046875, 7.755859375]
df.b=l
df
Out[190]: 
     a         b
0  NaN  5.000000
1  NaN  4.000000
2  NaN  6.000000
3  NaN  7.000000
4  1.0  6.500000
5  2.0  7.875000
6 -1.0  5.843750
7  0.0  6.804688
8  1.0  7.755859