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()