一步一步来:

这是分析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()

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐