python散点图中如何添加拟合线并显示拟合方程与R方?
polyfit()函数可以使用最小二乘法将一些点拟合成一条曲线.
numpy.polyfit(x, y, deg, rcond=None, full=False, w=None, cov=False)
# x:要拟合点的横坐标
# y:要拟合点的纵坐标
# deg:自由度.例如:自由度为2,那么拟合出来的曲线就是二次函数,自由度是3,拟合出来的曲线就是3次函数
首先我们先来构造一下需要被拟合的散点
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(-1, 1, 0.02)
y = 2 * np.sin(x * 2.3) + np.random.rand(len(x))
然后打印一下看看
plt.scatter(x, y)
plt.show()
然后用polyfit函数来把这些点拟合成一条3次曲线
parameter = np.polyfit(x, y, 3)
输出的结果为3次方程的参数,我们可以像下面这样把方程拼接出来
y2 = parameter[0] * x ** 3 + parameter[1] * x ** 2 + parameter[2] * x + parameter[3]
将拟合后的结果打印一下
plt.scatter(x, y)
plt.plot(x, y2, color='g')
plt.show()
还可以使用poly1d()函数帮我们拼接方程,结果是一样的
p = np.poly1d(parameter)
plt.scatter(x, y)
plt.plot(x, p(x), color='g')
plt.show()
评估指标R方
二维散点进行任意函数的最小二乘拟合
最小二乘中相关系数与R方的关系推导
其中,
利用相关系数矩阵计算R方
correlation = np.corrcoef(y, y2)[0,1] #相关系数
correlation**2 #R方
先来看下poly1d函数自带的输出结果
p = np.poly1d(parameter,variable='x')
print(p)
输出为
这里是把结果输出到两行里了,但是输出到两行是非常不方便的
尝试下自己编写函数,使输出到一行里
parameter=[-2.44919641, -0.01856314, 4.12010434, 0.47296566] #系数
aa=''
deg=3
for i in range(deg+1):bb=round(parameter[i],2) #bb是i次项系数if bb>=0:if i==0:bb=str(bb)else:bb=' +'+str(bb)else:bb=' '+str(bb)if deg==i:aa=aa+bbelse:aa=aa+bb+'x^'+str(deg-i)
print(aa)
输出为:
封装成函数
def Curve_Fitting(x,y,deg):parameter = np.polyfit(x, y, deg) #拟合deg次多项式p = np.poly1d(parameter) #拟合deg次多项式aa='' #方程拼接 ——————————————————for i in range(deg+1): bb=round(parameter[i],2)if bb>0:if i==0:bb=str(bb)else:bb='+'+str(bb)else:bb=str(bb)if deg==i:aa=aa+bbelse:aa=aa+bb+'x^'+str(deg-i) #方程拼接 ——————————————————plt.scatter(x, y) #原始数据散点图plt.plot(x, p(x), color='g') # 画拟合曲线# plt.text(-1,0,aa,fontdict={'size':'10','color':'b'})plt.legend([aa,round(np.corrcoef(y, p(x))[0,1]**2,2)]) #拼接好的方程和R方放到图例plt.show()
# print('曲线方程为:',aa)
# print(' r^2为:',round(np.corrcoef(y, p(x))[0,1]**2,2))
利用封装的函数重新画图
Curve_Fitting(x,y,3)
python散点图中如何添加拟合线并显示拟合方程与R方?相关推荐
- python散点图拟合曲线-python散点图:如何添加拟合线并显示拟合方程与R方?
原标题:python散点图:如何添加拟合线并显示拟合方程与R方? polyfit()函数可以使用最小二乘法将一些点拟合成一条曲线. numpy.polyfit(x, y, deg, rcond=Non ...
- 解决python绘图中中文及其他符号显示问题
解决python绘图中中文及其他符号显示问题,记录如下: import matplotlib.pyplot as plt# 设置显示中文 plt.rcParams['font.sans-serif'] ...
- python tkinter进度条_在python tkinter中Canvas实现进度条显示的方法
如下所示: from tkinter import * import time #更新进度条函数 def change_schedule(now_schedule,all_schedule): can ...
- R语言编写自定义函数、评估回归模型预测变量的相对重要性(Relative importance)、通过在所有可能的子模型中添加一个预测变量而获得的R方的平均增加、评估预测变量的重要度、并通过点图可视化
R语言编写自定义函数.评估回归模型预测变量的相对重要性(Relative importance).通过在所有可能的子模型中添加一个预测变量而获得的R方的平均增加.来评估预测变量的重要程度.并通过点图可 ...
- R语言使用lm构建线性回归模型、并将目标变量对数化(log10)实战:可视化模型预测输出与实际值对比图、可视化模型的残差、模型预测中系统误差的一个例子 、自定义函数计算R方指标和均方根误差RMSE
R语言使用lm构建线性回归模型.并将目标变量对数化(log10)实战:可视化模型预测输出与实际值对比图.可视化模型的残差.模型预测中系统误差的一个例子 .自定义函数计算R方指标和均方根误差RMSE 目 ...
- python pyplot中axis_Python Pyplot xaxis未显示在图形上
pyplot未在图形上显示x轴:import pandas as pd import matplotlib.pyplot as plt df = pd.read_csv('sitka_weather_ ...
- python散点图重叠_通过增加figure siz避免python散点图中的重叠点
我有以下几点:x = (0.5, 0.5, -3.6, -2, -2, -2.416, -2.75, -2.583, -2.583, -2.625, -2.541, -2.416, -2.416, - ...
- iOS 中如何添加应用自己的字体,使用萍方体
1.第一步找到你想用的字体的 ttf 格式.加入到你的工程的resouce目录下. 2.在工程的plist中AddRow,"Fonts provided by application&quo ...
- python字典怎么添加值_python字典中如何添加键值对
添加键值对 首先定义一个空字典 1 >>> dic={} 直接对字典中不存在的key进行赋值来添加 1 2 3 >>> dic['name']='zhangsan' ...
最新文章
- Pycharm 基本快捷键
- python肘部法则 最优分类
- 打造微量元素产业-丰收节交易会·李喜贵:签约南国健康产业
- leetcode 84. Largest Rectangle in Histogram | 84. 柱状图中最大的矩形(单调栈)
- 若依的框架怎么样_基于bootstrapTable的若依框架如何获取表格选中行的整行数据?...
- mongoose --- 建立一个集合规则,并导出.
- Linux开启fileinfo扩展
- 2016.08.19
- 【java机器学习】词向量在贝叶斯中的概念
- POJ 2240	Arbitrage Bellman_ford 判读是否存在正环
- 在CentOS 7上安装Node.js的4种方法(yum安装和源码安装)
- 创业公司如何做到零成本实现用户快速增长
- 解决nginx服务器显示txt文本为乱码的问题
- 带log的计算器html代码,lg计算器(log计算器在线)
- VBA控制Excel行列转换
- curry函数(柯里化)
- 后门程序技术知识全面深解
- 【PX4自动驾驶用户指南】距离传感器
- beego框架出现的问题-----panic: ./ippanichandle.exe flag redefined: graceful
- asan c/c++内存检测