工龄与薪水的一元线性回归模型-Python Pycharm实现（出现警告）

YYKKYY13579

已于 2022-10-06 14:08:45 修改

阅读量1.1k

点赞数 1

文章标签： python 线性回归 pycharm

于 2022-10-06 14:06:56 首次发布

本文链接：https://blog.csdn.net/YYKKYY13579/article/details/127182596

版权

import pandas
df = pandas.read_excel('IT行业收入表.xlsx')
df.head()

X = df[['工龄']]
Y = df['薪水']

from matplotlib import pyplot as plt
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.scatter(X, Y)
plt.xlabel('工龄')
plt.ylabel('薪水')
plt.show()

from sklearn.linear_model import LinearRegression
regr = LinearRegression()
regr.fit(X, Y)
import numpy as np
X = np.array(X)
plt.scatter(X, Y)
plt.plot(X, regr.predict(X), color='red')
plt.xlabel('工龄')
plt.ylabel('薪水')
plt.show()

两个图都运行出来了，但是不知道警告的原因？