PandasError：KeyError：“ ['Brain']不在索引中”

Question

这是代码：

data=pd.read_csv("/home/crpsm/Pycharm/DataSet/headbrain.csv")
print(data.describe())

y=data[["Brain"]]

x=data[["Head"]]

当我运行此代码时，出现此错误：

"['Brain'] not in index"

有什么办法可以解决此错误？ 在此先感谢

Answer 1

列名末尾有空格，需要str.rstrip或str.strip：

print(data.columns.tolist())
['Head ', 'Brain ']

data.columns = data.columns.str.rstrip() 

---

data.columns = data.columns.str.strip()