ValueError on sklearn的linear_model.predict

Question

import numpy as np 
import pandas as pd
from sklearn import datasets, linear_model
df = pd.read_csv("homeprices.csv")
df
model = linear_model.LinearRegression()
model.fit(df[['area']], df.price)
model.predict(5000)

ValueError:预期的2D数组，得到标量数组，而不是: array=5000。如果数据具有单个特性，则使用array.reshape(-1，1)或array.reshape(1，-1)对数据进行整形。

Answer 1

变更线：

model.predict(5000)

至：

model.predict([[5000]])