pywt

1
import matplotlib.pyplot as plt
2
import numpy as np
3

4
import pywt
5

6
wavlist = pywt.wavelist(kind="continuous")
7
cols = 3
8
rows = (len(wavlist) + cols - 1) // cols
9
fig, axs = plt.subplots(rows, cols, figsize=(10, 10),
10
                        sharex=True, sharey=True)
11
for ax, wavelet in zip(axs.flatten(), wavlist):
12
    # A few wavelet families require parameters in the string name
13
    if wavelet in ['cmor', 'shan']:
14
        wavelet += '1-1'
15
    elif wavelet == 'fbsp':
16
        wavelet += '1-1.5-1.0'
17

18
    [psi, x] = pywt.ContinuousWavelet(wavelet).wavefun(10)
19
    ax.plot(x, np.real(psi), label="real")
20
    ax.plot(x, np.imag(psi), label="imag")
21
    ax.set_title(wavelet)
22
    ax.set_xlim([-5, 5])
23
    ax.set_ylim([-0.8, 1])
24

25
ax.legend(loc="upper right")
26
plt.suptitle("Available wavelets for CWT")
27
plt.tight_layout()
28
plt.show()
29