多元回歸 model 使用 scikit 在 python 中學習

Question

當我使用自變量 'age'....reg.fit() 時顯示錯誤，例如“['age'] not in index”......有人可以幫幫我嗎？

import pandas as pd
import numpy as np
from sklearn import linear_model
df=pd.read_csv('homeprices_multivariate.csv')
import math
median_bedrooms=math.floor(df.bedrooms.median())
df.bedrooms=df.bedrooms.fillna(median_bedrooms)
reg=linear_model.LinearRegression()
reg.fit(df[['area','bedrooms','age']],df.price)

然后錯誤

['年齡'] 不在索引中

Answer 1

剝離列名，它應該可以工作：

df.columns=[c.strip() for c in df.columns]