kesmarag / stationary-wavelet-packet-transform Goto Github PK

import numpy as np
import matplotlib.pyplot as plt
from kesmarag.swpt import SWPT

signal = np.random.randn(1024,)
model = SWPT(max_level=3)
model.decompose(signal)
res = model.get_level(3)
'''
  level 1 => (0.A, 1.D)
  level 2 => (0.AA, 1.AD), (3.DA, 2.DD)
  level 3 => (0.AAA, 1.AAD), (3.ADA, 2.ADD), (7.DAA, 6.DAD), (4.DDA, 5.DDD)
'''
plt.pcolor(res)
plt.show()