请用python对excel表格数据进行线性回归输出回归系数、回归系数常数项、判定系数并绘制散点图
首先需要安装两个库:pandas和matplotlib。
安装命令:
!pip install pandas matplotlib
然后就可以用下面的代码进行线性回归,输出回归系数、回归系数常数项、判定系数并绘制散点图了。
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
from sklearn.metrics import r2_score
# 读取excel数据
data = pd.read_excel('data.xlsx')
# 提取x和y数据
x = data['x'].values.reshape(-1, 1)
y = data['y'].values.reshape(-1, 1)
# 创建线性回归模型
model = LinearRegression()
# 训练模型
model.fit(x, y)
# 输出回归系数和常数项
print('回归系数:', model.coef_[0][0])
print('常数项:', model.intercept_[0])
# 预测结果
y_pred = model.predict(x)
# 计算判定系数
r2 = r2_score(y, y_pred)
print('判定系数:', r2)
# 绘制散点图和回归线
plt.scatter(x, y)
plt.plot(x, y_pred, color='red')
plt.show()
其中,data.xlsx是带有x和y两列数据的Excel文件,可以根据实际情况进行修改
原文地址: http://www.cveoy.top/t/topic/hxi9 著作权归作者所有。请勿转载和采集!