是否有办法找到行与NaN，然后下降，以及n行数之后?

Question

我有一个电子表格，其中数据以一种奇怪的方式布局，数据块由单个NaN单元格的空行分隔。当然，我可以用drop.na删除NaN行，但是，是否有一种方法可以让我删除这些NaN行以及它们下面指定数量的行?

例如，我想删除数据框中的每个NaN行以及它下面的2行非NaN。

Answer 1

一种方法是从移位的列值创建掩码，例如:

# identify nan rows, true if nan
df['row_is_na'] = df['some_column'].isna()
# identify two rows that follow nan, true if after nan
df['rows_after_nan'] = df['row_is_na'].shift(-1) | df['row_is_na'].shift(-2)
# apply the mask
df = df[~df['rows_after_nan']]