flaredetector: add function to plot fits to an axis

2024-04-15 16:01:52 +02:00
parent 0627296727
commit b56a2e3701
1 changed files with 4 additions and 0 deletions
@@ -42,6 +42,10 @@ def plotFlarePeaks(ax, peaks):
    for peak in peaks:
        ax.plot(peak["FlarePeakTime"], peak["FlarePeak"], "x", color="red")

+def plotFlareFits(ax, fits):
+    for fit in fits:
+        ax.plot(fit["FlareFitTime"], fit["FlareFit"], "g--")
+
 def getFlareRange(flux, peakIndex):
    minInd = -np.inf
    maxInd = np.inf