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_)