diff --git a/script/fit.py b/script/fit.py index dcaa077..0e86111 100644 --- a/script/fit.py +++ b/script/fit.py @@ -10,18 +10,22 @@ def fit(ydata, xdata = None, draw_plot = True): print "Max index:" + str(index_max), print " x:" + str(max_x), print " y:" + str(max_y) - print "Server" gaussians = fit_gaussians(ydata, xdata, [index_max,]) (norm, mean, sigma) = gaussians[0] - print "Server" if draw_plot: + print "1" p = plot([ydata],["data"],[xdata], title="Fit" )[0] + print "2" fitted_gaussian_function = Gaussian(norm, mean, sigma) + print "3" scale_x = [float(min(xdata)), float(max(xdata)) ] + print "4" points = max((len(xdata)+1), 100) + print "5" resolution = (scale_x[1]-scale_x[0]) / points fit_y = [] fit_x = frange(scale_x[0],scale_x[1],resolution, True) + print "6" for x in fit_x: fit_y.append(fitted_gaussian_function.value(x)) if p is not None: