In [49]:
import numpy as np
import pyspeckit
from astropy.io import fits
import matplotlib.pyplot as plt
#% matplotlib.pyplot inlne
In [50]:
f_sum = "./test_crires.nod.ms.norm.sum.fits"
f_wavecal = "./test_crires.nod.ms.norm.sum.wavecal.fits"
In [51]:
data = fits.getdata(f_sum)
hdr = fits.getheader(f_sum)
print(data)
data_wavecal = fits.getdata(f_wavecal)
hdr_wavecal = fits.getheader(f_wavecal)
print(data_wavecal)
#data_wavecal.columns
wl = data_wavecal["Wavelength"]
flux = data_wavecal["Extracted_DRACS"]
In [65]:
# Test on data
xaxis = np.arange(len(data))
test_spec = pyspeckit.Spectrum(data=data, xarr=xaxis, unit='')
# xarrkwargs={'unit':'km/s'})
#sp = pyspeckit.Spectrum(data=data, error=error, xarr=xaxis,
# xarrkwargs={'unit':'km/s'},
# unit='erg/s/cm^2/AA')
test_spec.plotter()
In [67]:
test_spec.data
test_spec.unit
Out[67]:
In [69]:
# Test on wavelength calibtred data
test_spec2 = pyspeckit.Spectrum(data=flux, xarr=wl, xarrkwargs={'unit':'nm'}, unit='')
test_spec2.unit
test_spec.plotter()
In [ ]: