diff options
author | Blaise Thompson <blaise@untzag.com> | 2018-04-08 18:01:07 -0500 |
---|---|---|
committer | Blaise Thompson <blaise@untzag.com> | 2018-04-08 18:01:07 -0500 |
commit | e43de3cb79a9fad846babf0ddfdca21622b903dc (patch) | |
tree | 016582d5e0738ecf3754606101e18c1be413e057 /processing/fit_function.py | |
parent | cbc819350f29661b69a2ee6bf4f1dafcf3e2f621 (diff) |
2018-04-08 18:01
Diffstat (limited to 'processing/fit_function.py')
-rw-r--r-- | processing/fit_function.py | 21 |
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')) |