Pandas DataFrame选择列异常：无法选中索引内的'id'列

嘿，我猜你遇到的问题根源应该是**'id'是DataFrame的索引，不是普通列**！毕竟你说确认它在索引里，那直接用列名去取肯定会报错，但有些操作会自动带上索引，所以才会出现“有的代码可行有的不行”的情况。我给你几个实用的解决方案：

方法1：把索引转成普通列再选 #

最简单的办法就是先把'id'索引转成列，之后就能像选普通列一样操作了：

# 重置索引，把原索引'id'保留成列（drop=False是关键）
df = df.reset_index(drop=False)
# 现在直接选两列就行
selected_data = df[['id', 'text']]

方法2：不修改原表，直接组合索引和列 #

不想改动原DataFrame的话，可以单独提取索引，再和'text'列拼起来：

import pandas as pd

# 把索引转成Series，并重命名为'id'
id_col = df.index.rename('id').to_series()
# 横向拼接'id'和'text'
selected_data = pd.concat([id_col, df['text']], axis=1)

方法3：用.loc直接操作索引+列 #

如果只是需要查看或处理数据，也可以用.loc来同时指定索引和列，不用转列：

# 获取所有行的'text'列，此时结果会保留'id'索引
result = df.loc[:, ['text']]
# 如果需要把索引变成列显示，再加一步reset_index
result_with_id = result.reset_index()

为啥有的代码能跑通？ #

你说有的代码可行，大概率是那些操作默认会包含索引——比如用df.to_csv()的时候如果没设index=False，导出的文件里就会有'id'；或者某些可视化工具会自动把索引当成一列展示。而报错的代码应该是直接用df['id']去取，这时候因为'id'不是列，就会触发KeyError啦。

要是能给个具体报错信息或者数据样例，还能更精准地帮你调，但上面这几个方法应该能解决你的问题～

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