Open vmax and vmin access in yasa.plot_spectrogram()

[matiasandina]

It's not always useful, but sometimes you want to plot different channels on a common scale.
I think plot_spectrogram() could have vmax=None and vmin=None and, if supplied, they would overrule the normalization of the colormap.

It might be that this can be done once the spectrogram is but I'm not so fluent in matplotlib to do so.

[raphaelvallat]

Hi @matiasandina,

Good idea. It should be pretty straightforward to implement. Do you want to submit a PR? Or I'll work on it for the next release.

Thanks,
Raphael

[matiasandina]

I'm about to head out for vacation, I can take a look at it in 2-3 weeks at minimum. El mié, 8 jun 2022 a las 12:41, Raphael Vallat ***@***.***>) escribió:

…

Hi @matiasandina <https://github.com/matiasandina>, Good idea. It should be pretty straightforward to implement. Do you want to submit a PR? Or I'll work on it for the next release. Thanks, Raphael — Reply to this email directly, view it on GitHub <#73 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABZF2N4VC5ZXJIW66J4T4ULVODEJZANCNFSM5YDDEAMA> . You are receiving this because you were mentioned.Message ID: ***@***.***>

[matiasandina]

Turns out I needed it myself and it was way easier than what I thought. Will submit a PR soon

from io import BytesIO
import numpy as np
import requests
import local version of plot_spectrogram

r = requests.get('https://github.com/raphaelvallat/yasa/raw/master/notebooks/data_full_6hrs_100Hz_Cz%2BFz%2BPz.npz', stream=True)
npz = np.load(BytesIO(r.raw.read()))
data = npz.get('data')[0, :]
sf = 100
fig = plot_spectrogram(data, sf)

fig2 = plot_spectrogram(data, sf, vmin=-50, vmax=100)

Trying to test it for failure

fig3 = plot_spectrogram(data, sf, vmin=-50, vmax='asdf')

Results in

113 assert fmax < sf / 2, 'fmax must be less than Nyquist (sf / 2).'
    114 assert isinstance(vmin, (int, float, type(None))), 'vmin must be int, float, or None.'
--> 115 assert isinstance(vmax, (int, float, type(None))), 'vmax must be int, float, or None.'
    116 if vmin is not None:
    117   assert isinstance(vmax, (int, float)), 'vmax must be int or float if vmin is provided'