【Python】np.polyfit点拟合曲线

最近在学习python数据分析，做个笔记记录一下代码，今天是用numpy拟合了一组数据。

先贴拟合的效果出来，蓝色的点是用np生成的20个点，红线是拟合出来的曲线：

下面是代码：

import matplotlib.pyplot as plt
import numpy as npx = np.linspace(300,400,20)
y = x + np.random.random_integers(5,20,20) #随机取5到10中间20个数
poly = np.polyfit(x,y,deg=1)
z = np.polyval(poly, x)
plt.plot(x, y, 'o')
plt.plot(x, z)plt.show()

代码可以分成三部分：

第一部分是生成数据，x是在300到400中间平分的取了20个点（包含300，400）生成一个数组；至于y是在x的每个元素上加了一个5-20之间的随机数的数组。相当于获得了20个点的坐标。第二部分是用polyfit函数，对x,y进行拟合（下面会介绍这个函数）；拟合出的曲线对应的z值用polyval获得。第三部分就是绘图了。
下面详细介绍以下polyfit函数的参数以及返回值：这个函数前两个参数是x,y 也就是我们生成的两个20维的数组，代表要拟合的20个点。第三个参数1是你要拟合的多项式的最高阶阶数。比如你选择这个参数为1，即你想拟合一个有两项的多项式（最高阶为1），y=kx**1+bx**0,也就是一条直线，如果你选择参数为2（最高阶为2），就是拟合右边有三项的多项式：y=Lx**2+kx**1+bx**0,也就是一条抛物线。具体看下模块中的函数说明。

def polyfit(x, y, deg, rcond=None, full=False, w=None, cov=False):"""Least squares polynomial fit.Fit a polynomial ``p(x) = p[0] * x**deg + ... + p[deg]`` of degree `deg`to points `(x, y)`. Returns a vector of coefficients `p` that minimisesthe squared error.

至于polyfit的返回值，可以打印一下上面代码中的poly看一下，是[k,b]（如果你制定deg参数为2的话即返回抛物线的参数[L,k,b]）。

numpy.polyfit — NumPy v1.21 Manual

【Python】np.polyfit点拟合曲线相关推荐

numpy np.polyfit()（最小二乘多项式拟合曲线）（有待进一步研究）
from numpy\lib\polynomial.py @array_function_dispatch(_polyfit_dispatcher) def polyfit(x, y, deg, rc ...
python多项式拟合：np.polyfit 和 np.polyld
python数据拟合主要可采用numpy库,库的安装可直接用pip install numpy等. 这段代码可以直接用,但是要用自己的值 #多项式拟合 y = data_jiedian_2 #输入自己 ...
python求函数曲率_【Python】车道线拟合曲线的曲率半径计算公式及代码
学习优达学城的Advanced-Lane-Lines课程时,碰到了车道线的曲率半径计算.初见公式略显陌生,直到想起曲率半径的计算公式时才想明白,故记录如下. def cal_curverature(i ...
python根据坐标点拟合曲线绘图
python根据坐标点拟合曲线绘图任何程序错误,以及技术疑问或需要解答的,请扫码添加作者VX:1755337994 import os import numpy as np from scipy i ...
python中polyfit、poly1d函数
python中polyfit.poly1d函数一.polyfit函数函数原型 np.polyfit(x,y,num) 可以对一组数据进行多项式拟合下面举个例子说明: 其中:np.polyval( ...
【Python】车道线拟合曲线的曲率半径计算公式及代码
学习优达学城的Advanced-Lane-Lines课程时,碰到了车道线的曲率半径计算.初见公式略显陌生,直到想起曲率半径的计算公式时才想明白,故记录如下. def cal_curverature(i ...
python np.mod() 取模运算
python np.mod() 取模运算取模运算并非算术运算中的取余运算%,看代码发现使用取余计算的结果一直都对不上,为了打破这个误区,手动计算探索其内部原因呀~ python np.mod()的取 ...
np.polyfit()
np.polyfit((left_bottom[0], apex[0]), (left_bottom[1], apex[1]), 1) #x = (left_bottom[0], apex[0]) # ...
python hist直方图拟合曲线_详解用Python为直方图绘制拟合曲线的两种方法
直方图是用于展示数据的分组分布状态的一种图形,用矩形的宽度和高度表示频数分布,通过直方图,用户可以很直观的看出数据分布的形状.中心位置以及数据的离散程度等. 在python中一般采用matplotli ...

【Python】np.polyfit点拟合曲线

【Python】np.polyfit点拟合曲线相关推荐

最新文章

热门文章