使用Pandas删除列触发KeyError：['label']未在轴中找到，指定axis=1仍无效

错误原因分析 #

• 列名不存在：你指定的['label1', 'label2']中至少有一个列名不在当前DataFrame的列列表里，Pandas无法找到对应列执行删除操作，因此触发KeyError。
• inplace=True的误用：当设置inplace=True时，drop方法会直接修改原DataFrame，返回值为None。你执行data = data.drop(...)会把None赋值给data，导致后续操作异常，甚至可能让你误判删除操作的结果。

解决方法 #

1. 先确认目标列是否存在 #

删除前先检查列名的有效性，避免操作不存在的列：

# 查看DataFrame的所有列名
print(data.columns.tolist())

# 筛选出存在的目标列，找出缺失的列
target_cols = ['label1', 'label2']
existing_cols = [col for col in target_cols if col in data.columns]
missing_cols = [col for col in target_cols if col not in data.columns]

if missing_cols:
    print(f"以下列不存在，无法删除：{missing_cols}")
# 只对存在的列执行删除操作
data.drop(existing_cols, axis=1, inplace=True)

2. 正确使用inplace参数 #

有两种规范的写法，二选其一即可：

• 写法一：用inplace=True直接修改原数据，无需赋值

# 直接修改原DataFrame，不需要重新赋值给data
data.drop(['label1', 'label2'], axis=1, inplace=True)

• 写法二：不使用inplace=True，将返回的新DataFrame赋值给变量

# 返回删除列后的新DataFrame，赋值给data，原数据不会被修改
data = data.drop(['label1', 'label2'], axis=1)

3. 忽略不存在的列（可选） #

如果希望直接跳过不存在的列、避免报错，可以添加errors='ignore'参数：

# 存在的列会被删除，不存在的列自动忽略，不会触发KeyError
data.drop(['label1', 'label2'], axis=1, inplace=True, errors='ignore')

内容的提问来源于stack exchange，提问作者Aditya Raikwar