sklearn的linear_model.predict上的ValueError

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数组，而是scalar数组：array = 5000。如果数据具有单个要素，则使用array.reshape（-1,1）重新整形数据;如果数据包含单个样本，则使用array.reshape（1，-1）重新整形数据。

Answer 1

改变线：

model.predict(5000)

至：

model.predict([[5000]])