如何使用列表对象在Pandas中选择指定列？解决KeyError报错问题

你遇到的问题很典型——loc默认是按行索引来选取数据的，当你直接写data.loc[colsel]时，Pandas会把colsel里的列名当成行索引的标签去查找，自然找不到匹配项，所以抛出了KeyError。

正确的列选取方法 #

这里有几种可靠的方式来选取你需要的列：

1. 直接使用方括号（最简洁）

selected_data = data[colsel]

2. 使用loc的完整语法（明确指定行和列）
   loc的语法是data.loc[行选择器, 列选择器]，用冒号:表示选取所有行，后面跟上你的列名列表，就能精准选中目标列：

selected_data = data.loc[:, colsel]

3. 额外验证：确保列名完全匹配
   虽然你已经做了去空白处理，但可以再确认一下所有列名确实存在于DataFrame中：

# 检查所有列名是否都在DataFrame的列索引里
all_cols_exist = all(col in data.columns for col in colsel)
print(all_cols_exist)  # 正常情况下应该返回True

为什么原来的写法不行？ #

再明确一下逻辑：data.loc[colsel]等价于data.loc[colsel, :]，也就是选取行索引为colsel里值的所有列，而你的DataFrame行索引是默认的0、1、2...，Pandas找不到对应行标签，因此报错。

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