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

astrodatagui: FlareSummaryPlotGUI: add x label and properly sum sources

Browse Source
This commit is contained in:
SGCMarkus
2024-05-27 16:20:14 +02:00
parent 86b9c6db8f
commit 43e23853fc
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/astrodatagui/FlareSummaryPlotGUI.py
+9 -5
View File
@@ -132,6 +132,7 @@ class FlareSummaryPlotGUI(QWidget):
self.figureAxis.scatter(x[Unknownfilter], data[Unknownfilter]["pdcsapPeaksCount"], marker="x", color="gray", label="Unknown PDCSAP Count")
self.figureAxis.set_ylabel("Flare count")
self.figureAxis.set_xlabel("File Number")
self.figureAxis.legend()
self.figure.canvas.draw_idle()
@@ -146,11 +147,8 @@ class FlareSummaryPlotGUI(QWidget):
'pdcsapPeaks',
'sapFits',
'sapPeaks'])
data = data.groupby(["StarName", "Source", "SpType"],
as_index=False).agg({"sapPeaksCount": "sum",
"pdcsapPeaksCount": "sum"})
x = np.arange(start=1, stop=len(data)+1, step=1, dtype=int)
showSourceFilter = np.full(np.shape(x), False)
showSourceFilter = np.full(len(data), False)
if(self.cbKepler.isChecked()):
showKepler = data["Source"] == "Kepler"
@@ -162,6 +160,11 @@ class FlareSummaryPlotGUI(QWidget):
showTESS = data["Source"] == "TESS"
showSourceFilter |= showTESS
data = data[showSourceFilter].groupby(["StarName", "SpType"],
as_index=False).agg({"sapPeaksCount": "sum",
"pdcsapPeaksCount": "sum"})
x = np.arange(start=1, stop=len(data)+1, step=1, dtype=int)
if(self.cbSpTypeL.isChecked()):
Lfilter = data["SpType"].str.startswith("L")
Lfilter &= showSourceFilter
@@ -208,5 +211,6 @@ class FlareSummaryPlotGUI(QWidget):
self.figureAxis.scatter(x[Unknownfilter], data[Unknownfilter]["pdcsapPeaksCount"], marker="x", color="gray", label="Unknown PDCSAP Count")
self.figureAxis.set_ylabel("Flare count")
self.figureAxis.set_xlabel("Star Number")
self.figureAxis.legend()
self.figure.canvas.draw_idle()