diff --git a/generate_plots_MKGF.py b/generate_plots_MKGF.py index 9d561ab..7469edb 100644 --- a/generate_plots_MKGF.py +++ b/generate_plots_MKGF.py @@ -55,6 +55,21 @@ folderPath = f"../{date}/" #folderPath = f"G:/Meine Ablage/Masterthesis/{date}/" mkdir_p(folderPath) +# remove any data that has no period +data = data[(data["FitType"] == "sine") | (data["FitType"] == "poly")] + +# remove data with multiple minima/maxima present +filterArray = [] +for ind, row in data.reset_index().iterrows(): + if(len(row["pdcsapPeriodMinima"]) == len(row["pdcsapPeriodMaxima"]) and + len(row["pdcsapPeriodMinima"]) == 1): + filterArray.append(True) + else: + filterArray.append(False) + +filterArray = np.array(filterArray) +data = data[filterArray] + for comboLength in range(1, len(spType) + 1): for combo in itertools.combinations(spType, comboLength): for maxFlarePeak in [1.01, 1.05, 1.1, 1.25, 1.5]: