python数据分析之简单线性回归分析

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

以下为代码：

# 导入sklearn库下的linear_model类
from sklearn import linear_model
# 导入pandas库，别名为pd
import pandas as pd

filename = r'D:\test.xlsx'
# 读取数据文件
data = pd.read_excel(filename)

# 转化自变量数据为矩阵
x = data.iloc[:,:1].values
# 转化因变量数据为矩阵
y = data.iloc[:,2].values
# print(x)
# print(y)

# 调用线性回归函数
clf = linear_model.LinearRegression()
# 开始线性回归计算
clf.fit(x,y)
# 得到斜率
print(clf.coef_[0])
# 得到截距
print(clf.intercept_)