python画抛物线_在python中利用最小二乘拟合二次抛物线函数的方法

1、最小二乘也可以拟合二次函数

我们都知道用最小二乘拟合线性函数没有问题，那么能不能拟合二次函数甚至更高次的函数呢？答案当然是可以的。下面我们就来试试用最小二乘来拟合抛物线形状的的图像。

对于二次函数来说，一般形状为 f(x) = a*x*x+b*x+c，其中a,b,c为三个我们需要求解的参数。为了确定a、b、c，我们需要根据给定的样本，然后通过调整这些参数，知道最后找出一组参数a、b、c，使这些所有的样本点距离f(x)的距离平方和最小。用什么方法来调整这些参数呢？最常见的自然就是我们的梯度下降喽。

spicy库中有名为leastsq的方法，只需要输入一系列样本点，给出待求函数的基本形状，就可以针对上述问题求解了。

2、抛物线拟合源码

#!/usr/bin/env python

# coding:utf-8

import numpy as np

import matplotlib.pyplot as plt

from scipy.optimize import leastsq

# 待拟合的数据

X = np.array([1,2,3,4,5,6])

Y=np.array([9.1,18.3,32,47,69.5,94.8])

# 二次函数的标准形式

def func(params, x):

a, b, c = params

return a * x * x + b * x + c

# 误差函数，即拟合曲线所求的值与实际值的差

def error(params, x, y):

return func(params, x) - y

# 对参数求解

def slovePara():

p0 = [10, 10, 10]

Para = leastsq(error, p0, args=(X, Y))

return Para

# 输出最后的结果

def solution():

Para = slovePara()

a, b, c = Para[0]

print "a=",a," b=",b," c=",c

print "cost:" + str(Para[1])

print "求解的曲线是:"

print("y="+str(round(a,2))+"x*x+"+str(round(b,2))+"x+"+str(c))

plt.figure(figsize=(8,6))

plt.scatter(X, Y, color="green", label="sample data", linewidth=2)

# 画拟合直线

x=np.linspace(0,12,100) ##在0-15直接画100个连续点

y=a*x*x+b*x+c ##函数式

plt.plot(x,y,color="red",label="solution line",linewidth=2)

plt.legend() #绘制图例

plt.show()

solution()

上面的代码中，稍微注意的是如下几点：

1.func是待拟合的曲线的形状。本例中为二次函数的标准形式。

2.error为误差函数。很多同学会问不应该是最小平方和吗？为什么不是func(params, x) - y * func(params, x) - y？原因是名为lasts的方法中帮我们做了。看一下sklearn中源码的注释就知道什么情况了：

Minimize the sum of squares of a set of equations.

x = arg min(sum(func(y)**2,axis=0))

二次方的操作在源码中帮我们实现了。

3.p0里放的是a、b、c的初始值，这个值可以随意指定。往后随着迭代次数增加，a、b、c将会不断变化，使得error函数的值越来越小。

4.leastsq的返回值是一个tuple，它里面有两个元素，第一个元素是a、b、c的求解结果，第二个则为cost function的大小！

3.程序的最终结果与拟合曲线

程序最终的输出结果：

a= 2.06607141425 b= 2.5975001036 c= 4.68999985496

cost:1

求解的曲线是:

y=2.07x*x+2.6x+4.68999985496

最终的拟合曲线：

4、模拟其他曲线

leastsq函数除了可以模拟线性函数二次函数等多项式，还适用于任何波形的模拟。

比如方波：

def square_wave(x,p):

a, b, c, T = p

y = np.where(np.mod(x-b,T)

y = np.where(np.mod(x-b,T)>T/2, -1+c/a, y)

return a*y

比如高斯分布：

def gaussian_wave(x,p):

a, b, c, d= p

return a*np.exp(-(x-b)**2/(2*c**2))+d

只要将上面代码中的func换成对应的函数即可！

以上这篇在python中利用最小二乘拟合二次抛物线函数的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

python画抛物线_在python中利用最小二乘拟合二次抛物线函数的方法相关推荐

python求抛物线函数_在python中利用最小二乘拟合二次抛物线函数的方法
1.最小二乘也可以拟合二次函数我们都知道用最小二乘拟合线性函数没有问题,那么能不能拟合二次函数甚至更高次的函数呢?答案当然是可以的.下面我们就来试试用最小二乘来拟合抛物线形状的的图像. 对于二次函数 ...
python打乱数据集_在Keras中利用np.random.shuffle()打乱数据集实例
我就废话不多说了,大家还是直接看代码吧~ from numpy as np index=np.arange(2000) np.random.shuffle(index) print(index[0:2 ...
用python画明星_用Python画一颗特别的心送给她
[前言] 520刚过去了,大家有没有跟自己喜欢的对象在一起呢? 520 当天你还在送玫瑰花?巧克力?情侣套装?... 小编就想说:你能不能换点新意呢,这些操作已经普通的不能再普通了吧!就像别人来例假 ...
用python画圆锥_用python画一幅美瞳，今日份来自程序员的浪漫
如果说,眼睛是心灵的窗户,那么,美瞳就是心灵的彩窗,就像下图中这样. 而我们今天所要挑战的,就是用python画美瞳,而且是五分钟之内画三百副争奇斗艳.各领风骚.绝不重样的美瞳.作为这颗地球上最纯情的 ...
怎么用python画房子_用python画一个小房子
用python画一个小房子 2020年07月22日 | 萬仟网IT编程 | 我要评论如何用python画一个小房子?效果图如下:代码如下:import turtle# 前置p = turtle.Pe ...
用python画佩奇_使用python画个小猪佩奇的示例代码
基本原理选好画板大小,设置好画笔颜色.粗细,定位好位置,依次画鼻子.头.耳朵.眼睛.腮.嘴.身体.手脚.尾巴,完事儿. 都知道,Turtle 是 Python 内置的一个比较有趣味的模块,俗称&qu ...
用python画皇冠_用python做数字油画或者从一幅画学习风格，去画另一幅画
原博文 2018-08-30 17:12 − 1. 用python做数字油画模块: pillow 2. 从一幅画学习风格,去画另一幅画 http://pytorch.org/tutorials/ad ...
python画雪_用Python画一棵带音乐的雪夜圣诞树
本文我们用 Python 来画一棵带音乐效果的雪夜圣诞树,基本思路如下:用 Python 画一棵圣诞树作为背景图在圣诞树背景图中添加雪落效果及音乐下面来看一下具体实现. 首先,我们来画一棵圣诞树, ...
用python画风车_用Python画小女孩放风筝的示例
我就废话不多说了,直接上代码吧! # coding:utf-8 2import turtle as t 3import random 4# 画心 5def xin(): 6 def curvemove ...

python画抛物线_在python中利用最小二乘拟合二次抛物线函数的方法

python画抛物线_在python中利用最小二乘拟合二次抛物线函数的方法相关推荐

最新文章

热门文章