mbergmann/gpgrewe2024
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed plotting for integrals

Browse Source
This commit is contained in:
Diana 2024-10-23 15:35:47 +02:00
parent 0a3a8c59af
commit 9ea4e23bae
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
code/plot_functions.py
View File

@ -72,7 +72,7 @@ functions_path = r"C:\Users\diana\OneDrive - UT Cloud\Master\GPs\GP1_Grewe\Proje
sys.path.append(functions_path) sys.path.append(functions_path)
import useful_functions as u import useful_functions as u
def plot_highlighted_integrals(frequency, power, points, color_mapping, points_categories, delta = 2.5): 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. Plot the power spectrum and highlight integrals that exceed the threshold.
@ -82,12 +82,10 @@ def plot_highlighted_integrals(frequency, power, points, color_mapping, points_c
An array of frequencies corresponding to the power values. An array of frequencies corresponding to the power values.
power : np.array power : np.array
An array of power spectral density values. An array of power spectral density values.
exceeding_points : list points : list
A list of harmonic frequencies that exceed the threshold. A list of harmonic frequencies to check and highlight.
delta : float delta : float
Half-width of the range for integration around each point. Half-width of the range for integration around each point.
threshold : float
Threshold value to compare integrals with local mean.
color_mapping : dict color_mapping : dict
A dictionary mapping each category to its color. A dictionary mapping each category to its color.
points_categories : dict points_categories : dict
@ -111,10 +109,16 @@ def plot_highlighted_integrals(frequency, power, points, color_mapping, points_c
if valid: if valid:
# Define color based on the category of the point # 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') 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 # 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') 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}")
# Print out point and color
print(f"Integral around {point:.2f} Hz: {integral:.5e}, Color: {color}")
# Annotate the plot with the point and its color
ax.text(point, max(power) * 0.9, f'{point:.2f}', color=color, fontsize=10, ha='center')
# Define left and right boundaries of adjacent regions # Define left and right boundaries of adjacent regions
left_boundary = frequency[np.where((frequency >= point - 5 * delta) & (frequency < point - delta))[0][0]] left_boundary = frequency[np.where((frequency >= point - 5 * delta) & (frequency < point - delta))[0][0]]
right_boundary = frequency[np.where((frequency > point + delta) & (frequency <= point + 5 * delta))[0][-1]] right_boundary = frequency[np.where((frequency > point + delta) & (frequency <= point + 5 * delta))[0][-1]]
@ -132,3 +136,4 @@ def plot_highlighted_integrals(frequency, power, points, color_mapping, points_c
return fig return fig