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

Changed dianas plot function

Browse Source
This commit is contained in:
Diana 2024-10-24 19:38:07 +02:00
parent 528823f317
commit 3a1d7748ba
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
code/plot_functions.py
View File

@ -102,7 +102,7 @@ def plot_highlighted_integrals(ax, frequency, power, points, color_mapping, poin
-------
None
"""
ax.plot(frequency, power) # Plot power spectrum on the existing axes
ax.plot(frequency, power, color = "k") # Plot power spectrum on the existing axes
for point in points:
# Calculate the integral and local mean
@ -117,21 +117,23 @@ def plot_highlighted_integrals(ax, frequency, power, points, color_mapping, poin
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_category}')
# Print out point, category, and color
#ax.text(f"{point_category}: Integral: {integral:.5e}, Color: {color}")
ax.axvspan(point - delta, point + delta, color=color, alpha=0.2, label=f'{point_category}')
# Text with categories and colors
ax.text(1000, 5.8e-5, "AM", fontsize=10, color="green", alpha=0.2)
ax.text(1000, 5.6e-5, "Nyquist", fontsize=10, color="blue", alpha=0.2)
ax.text(1000, 5.4e-5, "EODf", fontsize=10, color="red", alpha=0.2)
ax.text(1000, 5.2e-5, "Stimulus frequency", fontsize=10, color="orange", alpha=0.2)
ax.text(1000, 5.0e-5, "EODf of awake fish", fontsize=10, color="purple", alpha=0.2)
ax.set_xlim([0, 1200])
ax.set_ylim([0, 6e-5])
ax.set_xlabel('Frequency (Hz)')
ax.set_ylabel('Power')
ax.set_title('Power Spectrum with Highlighted Integrals')
ax.set_title('Power Spectrum with highlighted Integrals')
# Apply float formatting to the y-axis
ax.yaxis.set_major_formatter(ticker.FuncFormatter(float_formatter))