def get_data(KIC):
"""
Retrieve Kepler timeseries from a remote data repository.
Fields are 'HJD','flux','e_flux','bkg','quarter'.
@param KIC: kic number or list of filenames
@type KIC: integer or list
@return: data, header
@rtype: recarray, dict
"""
times = []
flux = []
e_flux = []
background = []
quarter = []
if isinstance(KIC,str) or isinstance(KIC,int):
filenames = download_light_curve(KIC)
else:
filenames = KIC
for filename in filenames:
header = pyfits.getheader(filename)
data = fits.read2recarray(filename,ext='LIGHTCURVE')
times.append(data['TIME']+2454833.)
flux.append(data['SAP_FLUX'])
e_flux.append(data['SAP_FLUX_ERR'])
background.append(data['SAP_BKG'])
quarter.append(np.ones(len(data))*header['quarter'])
data = np.rec.fromarrays([np.hstack(times),np.hstack(flux),np.hstack(e_flux),
np.hstack(background),np.hstack(quarter)],
names=['HJD','flux','e_flux','bkg','quarter'])
return data,header
def read_mesa_fits(filename, ext=2):
"""
Read MESA FITS file containing tracks and summary.
@param filename: name of the file
@type filename: str
@param ext: extension to read
@type ext: int
@return: global parameters, local parameters
@rtype: dict, recarray
"""
data, header = fits.read2recarray(filename, ext=ext, return_header=True)
return header, data
def read_mesa_fits(filename,ext=2):
"""
Read MESA FITS file containing tracks and summary.
@param filename: name of the file
@type filename: str
@param ext: extension to read
@type ext: int
@return: global parameters, local parameters
@rtype: dict, recarray
"""
data,header = fits.read2recarray(filename,ext=ext,return_header=True)
return header,data
def get_data(KIC):
"""
Retrieve Kepler timeseries from a remote data repository.
Fields are 'HJD','flux','e_flux','bkg','quarter'.
@param KIC: kic number or list of filenames
@type KIC: integer or list
@return: data, header
@rtype: recarray, dict
"""
times = []
flux = []
e_flux = []
background = []
quarter = []
if isinstance(KIC, str) or isinstance(KIC, int):
filenames = download_light_curve(KIC)
else:
filenames = KIC
for filename in filenames:
header = pyfits.getheader(filename)
data = fits.read2recarray(filename, ext='LIGHTCURVE')
times.append(data['TIME'] + 2454833.)
flux.append(data['SAP_FLUX'])
e_flux.append(data['SAP_FLUX_ERR'])
background.append(data['SAP_BKG'])
quarter.append(np.ones(len(data)) * header['quarter'])
data = np.rec.fromarrays([
np.hstack(times),
np.hstack(flux),
np.hstack(e_flux),
np.hstack(background),
np.hstack(quarter)
],
names=['HJD', 'flux', 'e_flux', 'bkg', 'quarter'])
return data, header
