diff --git a/mind_the_gaps/lightcurves/fermilightcurve.py b/mind_the_gaps/lightcurves/fermilightcurve.py index eda7302..c77d44b 100644 --- a/mind_the_gaps/lightcurves/fermilightcurve.py +++ b/mind_the_gaps/lightcurves/fermilightcurve.py @@ -1,8 +1,8 @@ # @Author: Andrés Gúrpide # @Date: 05-02-2022 -# @Email: agurpidelash@irap.omp.eu +# @Email: agl1f22@soton.ac.uk # @Last modified by: agurpide -# @Last modified time: 28-02-2022 +# @Last modified time: 08-01-2026 import numpy as np import astropy.units as u from mind_the_gaps.lightcurves.gappylightcurve import GappyLightcurve @@ -12,6 +12,7 @@ class FermiLightcurve(GappyLightcurve): """ A class to create a GappyLightcurve based on a csv file containing only times, rates and errors. """ + def __init__(self, input_file): """ @@ -19,26 +20,24 @@ def __init__(self, input_file): ---------- input_file:str, Path to a csv file containing time, rates and errors - skip_header: int - How many rows to skip """ time, y, yerr = self.readdata(input_file) super().__init__(time.to(u.s).value, y, yerr) - def readdata(self, input_file): data = np.genfromtxt("%s" % input_file, names=True, delimiter=",") time_column = data.dtype.names[0] rate_column = data.dtype.names[1] - - if "MJD" in data: + if "MJD" in time_column.upper(): time = data[time_column] << u.d else: time = data[time_column] << u.s y = data[rate_column] - yerr = (np.abs(data["%s_err_neg" % rate_column]) + data["%s_err_pos" % rate_column]) / 2 + yerr = ( + np.abs(data["%s_err_neg" % rate_column]) + data["%s_err_pos" % rate_column] + ) / 2 return time, y, yerr