Python学习之一元线性回归

最新推荐文章于 2024-10-18 00:00:00 发布

aabbsndn

最新推荐文章于 2024-10-18 00:00:00 发布

阅读量108

点赞数 1

文章标签： python 学习线性回归

本文链接：https://blog.csdn.net/aabbsndn/article/details/141230255

版权

下载模块：pandas,seaborn,matplotlib,statsmodels

import pandas   ###表格处理
import seaborn   #绘图
import matplotlib.pyplot #显示画面
import  statsmodels.api  #计算线性回归

data = pandas.read_csv('D:/Tencent/数据/WeChat/WeChat Files/wxid_yqpsiebp6yyd22/FileStorage/File/2024-08/excel.csv')
seaborn.lmplot(x='Years', y='Salary', data=data, fit_reg=True)
fit = statsmodels.api.formula.ols('Salary ~ Years', data=data).fit()
print(fit.params)
matplotlib.pyplot.savefig('homework.pdf', format='pdf', bbox_inches='tight')
matplotlib.pyplot.show()