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
395

被折叠的 条评论
为什么被折叠?



