From cf9958a1af269b2db3e1851845a47ce5a8a76e19 Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Tue, 26 Mar 2024 13:32:57 +0100 Subject: [PATCH] flaredetector: util: findMaxIndices: set data for light curve objects to their flux --- main/flaredetector/util.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/main/flaredetector/util.py b/main/flaredetector/util.py index 49eb329..e97f8f5 100644 --- a/main/flaredetector/util.py +++ b/main/flaredetector/util.py @@ -2,10 +2,13 @@ import numpy as np from scipy.signal import find_peaks from lightkurve.periodogram import Periodogram +from lightkurve.lightcurve import LightCurve def findMaxIndices(lc, num=3, distance=100): if(isinstance(lc, Periodogram)): data = lc.power + elif(isinstance(lc, LightCurve)): + data = lc.flux peak_indices, peak_dict = find_peaks(data, height=(None, None), distance=distance) peak_heights = peak_dict["peak_heights"] maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]]