pandas用均值填充nan_python – 如何用pandas中的滚动平均值填充nan值

这应该工作：

input_data_frame[var_list]= input_data_frame[var_list].fillna(pd.rolling_mean(input_data_frame[var_list], 6, min_periods=1))

请注意,窗口为6,因为它包含NaN本身的值(不计入平均值).此外,其他NaN值不用于平均值,因此如果在窗口中找到的值少于5个,则根据实际值计算平均值.

例：

df = {'a': [1, 1,2,3,4,5, np.nan, 1, 1, 2, 3, 4, 5, np.nan] }

df = pd.DataFrame(data=df)

print df

a

0 1.0

1 1.0

2 2.0

3 3.0

4 4.0

5 5.0

6 NaN

7 1.0

8 1.0

9 2.0

10 3.0

11 4.0

12 5.0

13 NaN

输出：

a

0 1.0

1 1.0

2 2.0

3 3.0

4 4.0

5 5.0

6 3.0

7 1.0

8 1.0

9 2.0

10 3.0

11 4.0

12 5.0

13 3.0