判断Excel表格中两列数据的线性关系

import pandas as pd
import matplotlib.pyplot as plt

# 读取Excel数据,注意修改地址
data = pd.read_excel('data.xlsx')

# 将两列数据分别赋值给X和Y
X = data['Column1']
Y = data['Column2']

# 创建线性回归模型
from sklearn.linear_model import LinearRegression
from sklearn.metrics import r2_score

# 用数据训练模型
model = LinearRegression()
model.fit(X.values.reshape(-1, 1), Y.values.reshape(-1, 1))

# 打印模型的系数
print('Coefficients: ', model.coef_)

# 打印模型的截距
print('Intercept: ', model.intercept_)

# 用模型进行预测
Y_pred = model.predict(X.values.reshape(-1, 1))
r2 = r2_score(Y, y_pred)
print('R^2: ', r2)

# 绘制线性关系图
plt.scatter(X, Y, color='blue')
plt.plot(X, Y_pred, color='red', linewidth=2)

# 添加图例
plt.legend(['Data', 'Linear Regression'], loc='upper left')

# 添加标题和标签
plt.title('Linear Relationship Between Column1 and Column2')
plt.xlabel('Column1')
plt.ylabel('Column2')

# 显示图形
plt.show()

运行上述代码,会出现如下警告:

RuntimeWarning: Glyph 20540 missing from current font. font.set_text(s, 0.0, flags=flags)

出现上图错误是因为plt画图时找不到字体,导致所画出的图,横众坐标轴标题出现方框,无法正确显示字。

那么解决上述问题,只需要在plt.show()前添加以下两行代码:

plt.rcParams['font.sans-serif']=['FangSong']#坐标轴显示字体为‘仿宋’的中文标签
plt.rcParams['axes.unicode_minus']=False

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值