有没有办法找到带有 NaN 的行，然后删除它以及之后的 n 行？

Question

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

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

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']]