From e43de3cb79a9fad846babf0ddfdca21622b903dc Mon Sep 17 00:00:00 2001
From: Blaise Thompson <blaise@untzag.com>
Date: Sun, 8 Apr 2018 18:01:07 -0500
Subject: 2018-04-08 18:01

---
 processing/fit_function.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 processing/fit_function.py

(limited to 'processing/fit_function.py')

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'))
-- 
cgit v1.2.3