diff --git a/main/flaredetector/util.py b/main/flaredetector/util.py new file mode 100644 index 0000000..52163e2 --- /dev/null +++ b/main/flaredetector/util.py @@ -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 \ No newline at end of file