python直方图的拟合_用python拟合直方图

下面是一个使用scipy.optimize拟合非线性函数(如高斯函数)的示例，即使数据位于范围不好的直方图中，这样简单的平均估计也会失败。偏移常数也会导致简单的正态统计量失败(对于纯高斯数据，只需删除p[3]和c[3])。from pylab import *

from numpy import loadtxt

from scipy.optimize import leastsq

fitfunc = lambda p, x: p[0]*exp(-0.5*((x-p[1])/p[2])**2)+p[3]

errfunc = lambda p, x, y: (y - fitfunc(p, x))

filename = "gaussdata.csv"

data = loadtxt(filename,skiprows=1,delimiter=',')

xdata = data[:,0]

ydata = data[:,1]

init = [1.0, 0.5, 0.5, 0.5]

out = leastsq( errfunc, init, args=(xdata, ydata))

c = out[0]

print "A exp[-0.5((x-mu)/sigma)^2] + k "

print "Parent Coefficients:"

print "1.000, 0.200, 0.300, 0.625"

print "Fit Coefficients:"

print c[0],c[1],abs(c[2]),c[3]

plot(xdata, fitfunc(c, xdata))

plot(xdata, ydata)

title(r'$A = %.3f\ \mu = %.3f\ \sigma = %.3f\ k = %.3f $' %(c[0],c[1],abs(c[2]),c[3]));

show()

输出：A exp[-0.5((x-mu)/sigma)^2] + k

Parent Coefficients:

1.000, 0.200, 0.300, 0.625

Fit Coefficients:

0.961231625289 0.197254597618 0.293989275502 0.65370344131