python之简单线性回归分析

Posted 2020-12-31 才华充电中

使用sklearn库的linear_model.LinearRegression()，可以非常简单的进行线性回归分析

以下为代码：

 1 # 导入sklearn库下的linear_model类
 2 from sklearn import linear_model
 3 # 导入pandas库，别名为pd
 4 import pandas as pd
 5 
 6 filename = r‘D:	est.xlsx‘
 7 # 读取数据文件
 8 data = pd.read_excel(filename)
 9 
10 # 转化自变量数据为矩阵
11 x = data.iloc[:,:1].values
12 # 转化因变量数据为矩阵
13 y = data.iloc[:,2].values
14 # print(x)
15 # print(y)
16 
17 # 调用线性回归函数
18 clf = linear_model.LinearRegression()
19 # 开始线性回归计算
20 clf.fit(x,y)
21 # 得到斜率
22 print(clf.coef_[0])
23 # 得到截距
24 print(clf.intercept_)