一步一步来:

这是分析1992年到200年 国内生产总值x和财政收入y的数据。当知道某年的GDP是就可以预测财政收入了

1、导入相关包

 

2、读入数据并查看

3、导入模型,reshape(-1,1) 表示我也不知道几行,反正是一列!

 

 4、查看估计的参数和截距项

5、假设某年的国内生产总值gdp为238.5,预测y

6、作图。

 以下是全部代码:

import pandas as pd

from sklearn import linear_model

import matplotlib.pyplot as plt

data = pd.read_excel('data8-1.xlsx','Sheet1')

data
regr = linear_model.LinearRegression()
regr.fit(data['x'].reshape(-1,1), data['y']) # reshape(-1,1) 表示我也不知道几行,反正是一列!
a, b = regr.coef_, regr.intercept_
print(a,b)
GDP = 238.5
print (a*GDP+b)
print (regr.predict(GDP))
# 画图
# 1.真实的点
plt.scatter(data['x'], data['y'], color='blue')

# 2.拟合的直线
plt.plot(data['x'], regr.predict(data['x'].reshape(-1,1)), color='red', linewidth=4)

plt.show()