translating from scipy.signal.spectrogram()

[Florin Andrei]

I'm learning about wavelets and I've just discovered pywt.

If I need to extract the frequency components as a function of time, and build a heatmap of that (a "spectrogram"), I can comfortably use scipy.signal.spectrogram() - it's a little clunky and I have to do my own filtering, but it works. See this example where I start with the shape of the signal and I create successively more refined spectrograms:

https://github.com/FlorinAndrei/misc/blob/master/fft.ipynb

What would be the equivalent with pywt? I'm trying to get to the point where I could get results out of pywt that are at least as good.