1 files changed, 21 insertions, 0 deletions
diff --git a/processing/fit_function.py b/processing/fit_function.py
new file mode 100644
index 0000000..247b476
--- /dev/null
+++ b/processing/fit_function.py
@@ -0,0 +1,21 @@
+# import
+import os
+import numpy as np
+import matplotlib.pyplot as plt
+import WrightTools as wt
+from WrightTools import fit
+# define
+here = os.path.abspath(os.path.dirname(__file__))
+# noisey gaussian
+xi = np.linspace(-100, 100, 25)
+yi = 20*np.exp(-0.5*((xi-5)/20.)**2)
+yi = np.random.poisson(yi)
+plt.scatter(xi, yi)
+# fitted
+g = wt.fit.Gaussian()
+ps = g.fit(yi, xi)
+xi = np.linspace(-100, 100, 101)
+model = g.evaluate(ps, xi)
+# plot
+plt.plot(xi, model)
+wt.artists.savefig(os.path.join(here, 'fit_function.png'))