Merge pull request #24 from bmorris3/fix-issue-22

Adding axis labels
bmorris3 · Jul 26, 2018 · c3abea1 · c3abea1
2 parents 2ded8d1 + 9cc84d8
commit c3abea1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/aesop/spectra.py b/aesop/spectra.py
@@ -393,6 +393,8 @@ def fit_order(self, spectral_order, polynomial_order, plots=False):
             plt.plot(spectrum.wavelength,
                      np.polyval(fit_params,
                                 spectrum.wavelength - mean_wavelength))
+            plt.xlabel('Wavelength [{0}]'.format(spectrum.wavelength_unit))
+            plt.ylabel('Flux')
             plt.show()
         return fit_params