import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
years = np.arange(2009,2020)
years
sales = np.array([0.5,9.36,52,191,352,571,912,1207,1682.69,2135,2684])
sales
plt.scatter(years,sales,c = 'red',marker='*',s = 80)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/7cd0a4c14ea0596d91465b70dfeaf52a.png)
X = (years - 2008).reshape(-1,1)
X
y = sales
y
from sklearn.linear_model import LinearRegression
lr = LinearRegression(fit_intercept=True)
lr.fit(X,y)
w = lr.coef_[0]
b = lr.intercept_
display(w,b)
plt.scatter(years -2008,sales,c = 'red',marker='*',s = 80)
x = np.linspace(1,12,50)
plt.plot(x,w*x + b,c = 'green')
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/5f328b339c88764cb81316d3de73af03.png)
X2 = np.concatenate([X**2,X],axis= 1)
X2.shape
lr = LinearRegression(fit_intercept=True)
X2 = np.concatenate([X**2,X],axis= 1)
lr.fit(X2,y)
w1,w2 = lr.coef_
b = lr.intercept_
display(w1,w2,b)
plt.scatter(years -2008,sales,c = 'red',marker='*',s = 80)
x = np.linspace(1,12,50)
f = lambda x :w1*x**2 + w2*x + b
plt.plot(x,f(x),c = 'green')
print('2020年天猫双十一销量预测:',np.round(f(12),1))
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/295d35574ba0bc20885f52480e2cda2e.png)
lr = LinearRegression(fit_intercept=True)
X3 = np.concatenate([X**3,X**2,X],axis= 1)
lr.fit(X3,y)
w1,w2,w3 = lr.coef_
b = lr.intercept_
plt.scatter(years -2008,sales,c = 'red',marker='*',s = 80)
x = np.linspace(1,12,50)
f = lambda x :w1*x**3 + w2*x**2 + w3*x + b
plt.plot(x,f(x),c = 'green')
print('2020年天猫双十一销量预测:',np.round(f(12),1))
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/11441d65ddffadc9fc4d92f67bf192b2.png)