我找到并复制了此代码,以便从Finding the full width half maximum of a peak获取FWHM(第二个答案)。我下面的代码使用我自己的数据。生成的图表看起来不对,因为我的数据显示在一侧,而绿框则在另一侧。什么必须改变,以便我可以看到我的数据高斯?
import numpy as np, scipy.optimize as opt
from pylab import *
def gauss(x, p):
return 1.0/(p[1]*np.sqrt(2*np.pi))*np.exp(-(x-p[0])**2/(2*p[1]**2))
x = [6711.19873047, 6712.74267578, 6714.28710938, 6715.83544922, \
6717.38037109, 6718.92919922, 6720.47509766, 6722.02490234, \
6723.57128906, 6725.11767578, 6726.66845703, 6728.21630859, \
6729.76757812, 6731.31591797, 6732.86816406, 6734.41699219, \
6735.96630859, 6737.51953125, 6739.06933594, 6740.62353516, \
6742.17431641, 6743.72900391]
y = [20.86093712, 23.60984612, 23.079916, 18.17703056, 18.24843597, \
16.70049095, 19.48906136, 16.7509613, 19.09896088, 32.03007889, \
54.56513977, 58.76417542, 40.93075562, 24.77710915, 17.68757629, \
17.60736847, 18.89552498, 17.84486008, 17.49455452, 18.29696465, \
18.55847931, 19.26465797]
# Fit a guassian
p0 = [0,70]
errfunc = lambda p, x, y: gauss(x, p) - y # Distance to the target function
p1, success = opt.leastsq(errfunc, p0[:], args=(x, y))
fit_mu, fit_stdev = p1
FWHM = 2*np.sqrt(2*np.log(2))*fit_stdev
print "FWHM", FWHM
plot(x,y)
plot(x, gauss(x,p1),lw=3,alpha=.5, color='r')
axvspan(fit_mu-FWHM/2, fit_mu+FWHM/2, facecolor='g', alpha=0.5)
show()
答案 0 :(得分:2)
看起来最初的猜测p0
,您传递给scipy.optimize.leastsq()
就是问题所在。如果它不够接近(例如不在数据范围内),那么返回的解决方案将没有多大意义。
要获得合理的输出,您可以初始化p0 = [6730,70]
,这将产生以下图:
然而,这仍然不是正确答案,因为您已跳过original answer中出现的PDF规范化步骤。以下是适用于您的代码的规范化代码段(在您适应高斯之前添加它):
# Renormalize to a proper PDF
y /= ((max(x) - min(x)) / len(x)) * np.sum(y)
添加它会产生以下图表:
高斯拟合不是很紧,但至少它有一个合理的均值和标准差,正如我最初在下面描述的那样。
问题源于最小二乘函数p1
的输出,与p0
相同。
换句话说,假设结果均值为0
而不是np.mean(x) = 6727.45
,则该块将在完全不同的位置绘制。此外,根据329.67
的标准偏差,将块的宽度绘制为70
,而不是基于46.28
的{{1}}。