python - Pandas ffill/bfill 用于特定数量的观察

Question

我有以下数据框:

 id     indicator 
 1          NaN
 1          NaN
 1          1
 1          NaN
 1          NaN
 1          NaN

其实我还有好几个id。我现在的问题是，如何针对特定范围进行向前或向后填充，例如仅针对下一个/最后 2 个观察结果。我的数据框应该是这样的:

 id     indicator 
 1          NaN
 1          NaN 
 1          1
 1          1
 1          1
 1          NaN

我知道命令

df.groupby("id")["indicator"].fillna(value=None, method="ffill")         

但是，这会填充所有缺失值，而不仅仅是接下来的两个观察值。有人知道解决方案吗？

Answer 1

我认为DataFrameGroupBy.ffill或 DataFrameGroupBy.bfill使用 limit 参数更好:

df.groupby("id")["indicator"].ffill(limit=3)

df.groupby("id")["indicator"].bfill(limit=3)

示例:

#5 value is in the end of group, so only one value is filled 
df['filled'] = df.groupby("id")["indicator"].ffill(limit=2)
print (df)
    id  indicator  filled
0    1        NaN     NaN
1    1        NaN     NaN
2    1        1.0     1.0
3    1        NaN     1.0
4    1        NaN     1.0
5    1        NaN     NaN
6    1        NaN     NaN
7    1        NaN     NaN
8    1        4.0     4.0
9    1        NaN     4.0
10   1        NaN     4.0
11   1        NaN     NaN
12   1        NaN     NaN
13   2        NaN     NaN
14   2        NaN     NaN
15   2        1.0     1.0
16   2        NaN     1.0
17   2        NaN     1.0
18   2        NaN     NaN
19   2        5.0     5.0
20   2        NaN     5.0
21   3        3.0     3.0
22   3        NaN     3.0
23   3        NaN     3.0
24   3        NaN     NaN
25   3        NaN     NaN