flaredetector: add function to plot fit peaks to an axis

2024-04-15 16:00:21 +02:00
parent af824867fb
commit 0627296727
1 changed files with 4 additions and 0 deletions
@@ -38,6 +38,10 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05

    return peaks, fits

+def plotFlarePeaks(ax, peaks):
+    for peak in peaks:
+        ax.plot(peak["FlarePeakTime"], peak["FlarePeak"], "x", color="red")
+
 def getFlareRange(flux, peakIndex):
    minInd = -np.inf
    maxInd = np.inf