import matplotlib.pyplot as plt
import numpy as np
from spectractor.extractor.spectroscopy import *
f, ax = plt.subplots(1,1)
ax.set_xlim(300,1000)
lines = Lines(HYDROGEN_LINES+ATMOSPHERIC_LINES)
lines.lines[5].fitted = True
lines.lines[5].high_snr = True
lines.lines[-1].fitted = True
lines.lines[-1].high_snr = True
ax = lines.plot_atomic_lines(ax)
plt.show()