From 86272cc18ac05d725026a0fabd1f2f84f808e8fb Mon Sep 17 00:00:00 2001 From: Diana Date: Wed, 23 Oct 2024 11:59:29 +0200 Subject: [PATCH] changed Dianas Plot function --- code/plot_functions.py | 24 ++++++++++++++++-------- 1 file changed, 16 insertions(+), 8 deletions(-) diff --git a/code/plot_functions.py b/code/plot_functions.py index 4d78cdd..63be604 100644 --- a/code/plot_functions.py +++ b/code/plot_functions.py @@ -6,6 +6,7 @@ import matplotlib.pyplot as plt import numpy as np import rlxnix as rlx from useful_functions import power_spectrum +import sys '''IMPORT DATA''' datafolder = '../data' #./ wo ich gerade bin; ../ eine ebene höher; ../../ zwei ebenen höher @@ -66,8 +67,12 @@ def power_spectrum_plot(f, p): ax.set_xlim(0, 1000) plt.show() -####### ADD DIANAS POWER SPECTRUM PLOT -def plot_highlighted_integrals(frequency, power, exceeding_points, delta, threshold, color_mapping, points_categories): +# DIANAS POWER SPECTRUM PLOT +functions_path = r"C:\Users\diana\OneDrive - UT Cloud\Master\GPs\GP1_Grewe\Projekt\gpgrewe2024\code" +sys.path.append(functions_path) +import useful_functions as u + +def plot_highlighted_integrals(frequency, power, points, color_mapping, points_categories, delta = 2.5): """ Plot the power spectrum and highlight integrals that exceed the threshold. @@ -96,16 +101,19 @@ def plot_highlighted_integrals(frequency, power, exceeding_points, delta, thresh fig, ax = plt.subplots() ax.plot(frequency, power) # Plot power spectrum - for point in exceeding_points: - integral, local_mean = calculate_integral(frequency, power, point, delta) - valid, _ = valid_integrals(integral, local_mean, threshold, point) + for point in points: + # Use the imported function to calculate the integral and local mean + integral, local_mean, _ = u.calculate_integral(frequency, power, point) + + # Use the imported function to check if the point is valid + valid = u.valid_integrals(integral, local_mean, point) + if valid: # Define color based on the category of the point color = next((c for cat, c in color_mapping.items() if point in points_categories[cat]), 'gray') # Shade the region around the point where the integral was calculated ax.axvspan(point - delta, point + delta, color=color, alpha=0.3, label=f'{point:.2f} Hz') print(f"Integral around {point:.2f} Hz: {integral:.5e}") - # Define left and right boundaries of adjacent regions left_boundary = frequency[np.where((frequency >= point - 5 * delta) & (frequency < point - delta))[0][0]] @@ -114,8 +122,7 @@ def plot_highlighted_integrals(frequency, power, exceeding_points, delta, thresh # Add vertical dashed lines at the boundaries of the adjacent regions ax.axvline(x=left_boundary, color="k", linestyle="--") ax.axvline(x=right_boundary, color="k", linestyle="--") - - + ax.set_xlim([0, 1200]) ax.set_xlabel('Frequency (Hz)') ax.set_ylabel('Power') @@ -124,3 +131,4 @@ def plot_highlighted_integrals(frequency, power, exceeding_points, delta, thresh return fig +