如何在Pandas中安全删除多列:不存在列不触发报错
解决方案
直接使用pandas.DataFrame.drop方法的errors参数并设置为'ignore',就能原生实现“仅删除存在的列,不存在的列自动忽略且不报错”的需求,无需额外写循环或自定义函数。
示例代码
df.drop(columns=['col1', 'col2'], inplace=True, errors='ignore')
参数说明
errors='ignore':当传入的列名中有不存在于DataFrame的项时,不会触发KeyError异常,只会删除那些实际存在的列,不存在的列直接被忽略。
举个例子:如果你的DataFrame里只有col2存在、col1不存在,执行上述代码后只会删除col2,程序不会报错。
内容的提问来源于stack exchange,提问作者John s
相关产品推荐
相关产品推荐

