test.py 557 B

123456789101112131415161718192021
  1. import pandas as pd
  2. import numpy as np
  3. # 创建一个示例DataFrame
  4. df = pd.DataFrame({
  5. 'A': [1, np.nan, 3],
  6. 'B': [np.nan, np.nan, np.nan],
  7. 'C': [2, 2, np.nan],
  8. 'D': [np.nan, 3, 3]
  9. })
  10. # 指定你想要检查的列
  11. columns_to_check = ['A', 'B', 'C']
  12. # 检查指定列中非全为空的列
  13. non_empty_columns = df[columns_to_check].apply(lambda x: x.notnull().any(), axis=0)
  14. # 获取非全为空的列名
  15. non_empty_column_names = non_empty_columns[non_empty_columns].index.tolist()
  16. print("非全为空的列名:", non_empty_column_names)