SGCMarkus/flaredetector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

flaredetector: util: set data per different instance specific to not duplicate code

Browse Source
This commit is contained in:
SGCMarkus
2024-03-26 12:40:12 +01:00
parent e5db328b9a
commit cbfbd7e976
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/flaredetector/util.py
+5 -4
View File
@@ -5,7 +5,8 @@ from lightkurve.periodogram import Periodogram
def findMaxIndices(lc, num=3, distance=100): def findMaxIndices(lc, num=3, distance=100):
if(isinstance(lc, Periodogram)): if(isinstance(lc, Periodogram)):
peak_indices, peak_dict = find_peaks(lc.power, height=(None, None), distance=distance) data = lc.power
peak_heights = peak_dict["peak_heights"] peak_indices, peak_dict = find_peaks(data, height=(None, None), distance=distance)
maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]] peak_heights = peak_dict["peak_heights"]
return maxIndices maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]]
return maxIndices