问题描述
我一直在尝试将高斯分布到我的频谱中。 (强度v/s速度谱)
我使用以下代码将数据拟合为高斯分布。但是,从结果中可以看出,拟合中只包含一个数据点。有什么我可以做的,以便我可以在高斯中包含更多点。
from numpy import exp,linspace,random
import matplotlib.pyplot as plt
def gaussian(x,amp,cen,wid):
return amp * exp(-(x-cen)**2 / wid)
from scipy.optimize import curve_fit
x = veLocity
y = data
print(x)
print(y)
init_vals = [0.00950554,60000,35] # for [amp,wid]
best_vals,covar = curve_fit(gaussian,x,y,p0=init_vals)
print(best_vals)
print(repr(covar))
ym = gaussian(x,best_vals[0],best_vals[1],best_vals[2])
fig = plt.figure()
ax = fig.add_subplot(211)
ax.plot(x,y)
ax = fig.add_subplot(212)
ax.plot(x,c='k',label='Function')
ax.plot(x,ym,c='r',label='Best fit')
plt.legend()
plt.show()
数据点:
x: [-5.99999993e+04 -4.99999993e+04 -3.99999993e+04 -2.99999993e+04
-1.99999993e+04 -9.99999934e+03 6.65010004e-04 1.00000007e+04
2.00000007e+04 3.00000007e+04 4.00000007e+04 5.00000007e+04
6.00000007e+04 7.00000007e+04 8.00000007e+04 9.00000007e+04
1.00000001e+05 1.10000001e+05 1.20000001e+05 1.30000001e+05
1.40000001e+05]
y: [ 0.00056511 -0.00098584 -0.00325616 -0.00101042 0.00168894 -0.00097406
-0.00134408 0.00128847 -0.00111633 -0.00151621 0.00299326 0.00916455
0.00960554 0.00317363 0.00311124 -0.00080881 0.00215932 0.00596419
-0.00192256 -0.00190138 -0.00013216]
这些是光谱的数据点。任何人都可以帮助我更好地拟合数据。我一直在尽我所能。
非常感谢。