Python实现小波包变换并对分量分别重构

工作中遇到需要对信号量进行“小波包变换”求取分量的需求，在此记录完成代码作为笔记。

1、下载外部包

进行小波包变换需要引入外部包PyWavelets和Matplotlib(Matplotlib主要用于对信号数据绘图以便方便查看)。 

pip install PyWavelets
pip install Matplotlib

2、引入依赖

import matplotlib.pyplot as plt
import pywt

3、小波包分解

tr = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
wp = pywt.WaveletPacket(data=tr, wavelet='db4', mode='symmetric', maxlevel=2)

 pywt.WaveletPacket函数进行小波包分解；

tr是原始信号值；wavelet='db4'表示使用db4作为小波基；mode='symmetric'表示数据处理模式；maxleve=2表示分解层数为两层。

4、获取分量数据

在获取分量之前可以查看分解结果中每个分量的系数（也就是通过这个系数从分解结果中获取分量数据）

# 根据频段频率（freq）进行排序
print([node.path for node in wp.get_level(1, 'freq')])
print([node