如何删除含有N个连续NaN值的列？

假设我们有一个包含NaN值的DataFrame：

import pandas as pd
import numpy as np

df = pd.DataFrame({'A': [1, 2, np.nan, 4, 5],
                   'B': [6, np.nan, np.nan, 9, 10],
                   'C': [11, 12, np.nan, np.nan, np.nan],
                   'D': [16, 17, 18, np.nan, np.nan]})

应该先计算每列中连续NaN值的数量：

consecutive_nans = df.isnull().astype(int).groupby(
        df.notnull().astype(int).cumsum()).cumsum()

接下来，可以找到那些具有多于N个连续NaN值的列：

N = 2 # 设置为2
cols_to_keep = ~(consecutive_nans > N).any()

最后，可以通过使用布尔索引来删除这些列：

df = df.loc[:, cols_to_keep]

最终的结果是：

    A     B     D
0  1.0   6.0  16.0
1  2.0   NaN  17.0
2  NaN   NaN  18.0
3  4.0   9.0   NaN
4  5.0  10.0   NaN