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

Changes dianas plot

Browse Source
This commit is contained in:
Diana 2024-10-28 15:19:37 +01:00
parent 851857c19d
commit 861a689f4e
1 changed files with 22 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
code/plot_functions.py
View File

@ -73,12 +73,12 @@ sys.path.append(functions_path)
import useful_functions as u
import matplotlib.ticker as ticker
import matplotlib.patches as mpatches
import matplotlib.cm as cm
def float_formatter(x, _):
"""Format the y-axis values as floats with a specified precision."""
return f'{x:.5f}'
def plot_highlighted_integrals(ax, frequency, power, points, nyquist, true_eodf, color_mapping, points_categories, delta=2.5):
"""
Highlights integrals on the existing axes of the power spectrum for a given dataset.
@ -104,8 +104,17 @@ def plot_highlighted_integrals(ax, frequency, power, points, nyquist, true_eodf,
-------
None
"""
# Define color mappings for specific categories
category_colors = {
"AM": "#ff7f0e",
"Nyquist": "#2ca02c",
"EODf": "#d62728",
"Stimulus": "#9467bd",
"EODf (awake fish)": "#8c564b"
}
# Plot the power spectrum on the provided axes
ax.plot(frequency, power, color="k")
for point in points:
# Identify the category for the current point
@ -120,20 +129,24 @@ def plot_highlighted_integrals(ax, frequency, power, points, nyquist, true_eodf,
if valid:
# Highlight valid points with a shaded region
ax.axvspan(point - delta, point + delta, color=color, alpha=0.2, label=f'{point_category}')
ax.axvspan(point - delta, point + delta, color=color, alpha=0.35, label=f'{point_category}')
ax.plot(frequency, power)
# Use the category colors for 'Nyquist' and 'EODf' lines
ax.axvline(nyquist, color=category_colors.get("Nyquist", "#2ca02c"), linestyle="--")
ax.axvline(true_eodf, color=category_colors.get("EODf (awake fish)", "#8c564b"), linestyle="--")
# Set plot limits and labels
ax.set_xlim([0, 1200])
ax.set_ylim([0, 6e-5])
ax.axvline(nyquist, color = "k", linestyle = "--")
ax.axvline(true_eodf, color = "k", linestyle = "--")
ax.set_xlabel('Frequency (Hz)')
ax.set_ylabel('Power')
ax.set_title('Power Spectrum with Highlighted Integrals')
ax.set_xlabel('Frequency (Hz)', fontsize=12)
ax.set_ylabel(r'Power [$\frac{\mathrm{mV^2}}{\mathrm{Hz}}$]', fontsize=12)
#ax.set_title('Power Spectrum with highlighted Integrals', fontsize=14)
# Apply float formatting to the y-axis
ax.yaxis.set_major_formatter(ticker.FuncFormatter(float_formatter))
#ax.legend(loc="upper right")