28 lines
1016 B
Python
28 lines
1016 B
Python
"""
|
|
Function fitting and peak search with mathutils facade
|
|
"""
|
|
from mathutils import fit_gaussian, Gaussian
|
|
from plotutils import plot_function, plot_data
|
|
import math
|
|
|
|
|
|
def plot_fit(xdata, ydata, gauss_pars=None, title = "Fit"):
|
|
if gauss_pars is None:
|
|
gauss_pars = fit_gaussian(ydata, xdata)
|
|
|
|
(normalization, mean_value, sigma) = gauss_pars
|
|
fitted_gaussian_function = Gaussian(normalization, mean, sigma)
|
|
print (normalization, mean_value, sigma)
|
|
|
|
p = plot(None, title=title)[0]
|
|
p.clear()
|
|
plot_data(p, ydata, "Data", xdata=xdata, show_points = True, color=Color.BLUE)
|
|
fit_range = frange(xdata[0],xdata[-1],float(xdata[1]-xdata[0])/100, True)
|
|
plot_function(p, fitted_gaussian_function, "Fit", fit_range, show_points=False, color=Color.RED)
|
|
p.setLegendVisible(True)
|
|
p.addMarker(mean_value, None, "Mean=" + str(round(mean_value,2)), Color.LIGHT_GRAY)
|
|
|
|
|
|
ydata = [0, 1,2,3,4,5,6,7,8,9,8,7,6,5,4,4,3,3,2,2,1,0]
|
|
xdata = range(len(ydata))
|
|
plot_fit(xdata, ydata) |