flaredetector: util: add function to find indices of peaks

2024-03-26 11:28:04 +01:00
parent 248a8c2497
commit ca2d09f932
1 changed files with 8 additions and 0 deletions
@@ -0,0 +1,8 @@
+import numpy as np
+from scipy.signal import find_peaks
+
+def findMaxIndices(data, num=3, distance=100):
+    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]]
+    return maxIndices