有没有一种方法可以用NaN找到行，然后删除它以及n行之后的行？

Question

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

例如，我想删除dataframe中的每个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']]