Python Pandas删除指定列遇KeyError：列索引未找到的解决求助

错误原因分析 #

你的代码存在两个核心问题：

1. 循环逻辑混乱：嵌套循环完全没必要，而且每次执行df.drop(..., inplace=True)后，DataFrame的列数会实时减少，但循环仍基于最初的列数范围执行，导致后续尝试删除已经被移除的列，触发KeyError。
2. 变量未定义：代码中使用了colList但没看到你定义它，假设它是你要保留的[1,2,3]，但双层循环的判断逻辑会重复执行删除操作，进一步加剧错误。

正确实现方式 #

方法1：直接筛选保留列（最推荐，简洁高效）

不需要删除列，直接选择你需要的列即可：

df = df[[1, 2, 3]]

如果需要原地修改（不创建新的DataFrame），可以用：

df.drop(df.columns.difference([1,2,3]), axis=1, inplace=True)

方法2：批量删除不需要的列

先找出所有需要删除的列，一次性批量删除，避免循环中多次修改DataFrame导致的错误：

# 生成所有不在保留列表中的列
cols_to_drop = [col for col in df.columns if col not in [1, 2, 3]]
# 批量删除
df.drop(cols_to_drop, axis=1, inplace=True)

额外优化：列名替换的简化 #

你原来替换列名的代码可以简化成一行，不用循环：

df.columns = range(len(df.columns))

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