如何修复pandas删除列时出现的not found in axis KeyError报错

axis=1只是告诉pandas你要删的是列、不是行，根本不负责校验你写的列名是否真实存在。报这个KeyError和你指不指定axis参数没有关系，纯粹是你传入的两个待删列名['unnamed: 31','short name']和DataFrame实际存储的列索引无法完全匹配。
常见的匹配失败场景：

• 大小写不匹配：比如pandas读文件自动生成的未命名列默认格式是Unnamed: 31（首字母大写），传入全小写的名称会直接判定为不存在
• 列名携带隐藏的前后空格：比如数据读入时列名尾部带看不见的空格，实际值为'short name '，肉眼无法识别但匹配时会直接失败
• 目标列已经在之前的操作中被删除，重复执行drop代码时自然找不到对应列
• 列名存在符号差异：比如冒号为全角格式、携带不可见的转义特殊字符，都会导致匹配失败

• 先打印全量列名核对真实取值，执行如下代码即可：

  print(df.columns.tolist())
  

  对照输出的列名列表，把待删列名完全替换为列表内的对应取值，不要凭记忆手写列名，能解决绝大多数这类报错。
• 如果不需要强制校验列是否存在，只是要实现“列存在就删除、不存在也不中断流程”的效果，可以给drop方法加errors='ignore'参数，自动跳过不存在的列：

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

• 如果批量处理数据时经常遇到列名带空格、大小写不统一的问题，可以先统一清洗列名再执行删除操作，从根源规避格式类匹配问题：

  # 去除所有列名前后空白符、统一转为小写
  df.columns = df.columns.str.strip().str.lower()
  # 清洗完成后再执行删除
  df.drop(['unnamed: 31','short name'], axis=1, inplace=True)
  

内容的提问来源于stack exchange，提问作者Iqra Imtiaz