获取带有NaN的行以及NaN行的前后行

使用 boolean indexing 链质量为 | 用于逐位 OR 以下为:

m = df['location'].isna()

df = df[m.shift(fill_value=False) | m.shift(-1, fill_value=False) | m]
print (df)
   location                 time
0   Seattle  2022-06-01 12:00:00
1       NaN  2022-06-01 13:00:00
2  Portland  2022-06-01 14:00:00

我会使用居中 rolling 为生成掩码 boolean indexing 以下为:

N = 1
m = (df['location'].isna()
      .rolling(2*N+1, min_periods=1, center=True)
      .max().eq(1)
     )

out = df.loc[m]

您可以通过更改 N 。

输出:

   location                 time
0   Seattle  2022-06-01 12:00:00
1       NaN  2022-06-01 13:00:00
2  Portland  2022-06-01 14:00:00