如果df不包含特定的列名，请执行一些操作

Question

我想基于if else中不存在特定列名的情况来执行df操作。

if a_specific_column_is_NOT_in_the_df:
   print('not ok')
else:
   print('ok')

使用以下代码，我可以完成与任务相反的操作。

if [col for col in df.columns if 'A' in col]:
    print('ok')
else:
    print('not ok')

我的任务是，如果df 不不包含列名A，则打印not ok，否则，打印“ ok”。

谢谢！

Answer 1

只需在列表前面添加

即可

 if not [col for col in df.columns if 'A' in col]:
        print('not ok')
 else:
        print('ok')

例如，这将导致不正常：

if not [col for col in ['B','C'] if 'A' in col]:
    print('not ok')
else:
    print('ok')