如何修复删除列时在轴中找不到 [] 的键错误，如何用 pandas 修复此问题？

Question

我用了drop方法并使用了

df.drop(['unnamed: 31','short name'], axis=1, inplace=True)

我收到以下关键错误

“在轴中找不到['未命名：31''短名称']”。

为什么我定义轴时也是这样。

Answer 1

检查列列表以查看它们是否存在于数据框中。这是执行此操作的片段

# Create a dataframe
df = pd.DataFrame({"A" :[1,2,3],"B" :[343,32,232]})

# Get list of columns of df
COLUMNS = df.columns.values

# Define columns you want to delete, in your case it
COLUMNS_TO_DELETE = ["B","C","D"]

# Delete only those columns which are present in dataframe
COLUMNS_TO_DELETE = [col for col in COLUMNS_TO_DELETE if col in COLUMNS]

# Apply df.drop method
df.drop(columns=COLUMNS_TO_DELETE,axis=1)

Answer 2

我有同样的问题，刚刚测试了 @Mustafa Khalid 的建议，效果很好。现在我想知道为什么它不能使用初始代码，尽管它是相同的逻辑？