要在熊猫中删除列，但如果列已存在，则不应显示错误

Question

我想删除大熊猫中的列，但如果不存在则显示错误，我希望它不显示错误并删除存在的剩余列。是否有函数或任何方法可以帮助我实现那个。

Answer 1

在DataFrame.drop中使用参数errors='ignore'：

  

错误：{'ignore'，'raise'}，默认为'raise'
  如果为“忽略”，则抑制错误并仅删除现有标签。

df = pd.DataFrame({
    'A': ['a','b','c'],
    'B': list(range(3))
})
print (df)
   A  B
0  a  0
1  b  1
2  c  2

df = df.drop(['B','C'], errors='ignore', axis=1)
print (df)
   A
0  a
1  b
2  c