python小波包代码

以下是一个基于Python实现的小波包代码示例：

import numpy as np
import pywt

# 定义小波包分解函数
def wavelet_packet_decomposition(data, wavelet, level):
    wp = pywt.WaveletPacket(data, wavelet, 'symmetric', maxlevel=level)
    nodes = wp.get_level(level, 'natural')
    return np.array([n.data for n in nodes])

# 定义小波包重构函数
def wavelet_packet_reconstruction(data, wavelet, level):
    wp = pywt.WaveletPacket(None, wavelet, 'symmetric', maxlevel=level)
    for i in range(len(data)):
        node = 'a' + ''.join(['{}'.format(j) for j in wp[i].path])
        wp[node].data = data[i]
    return wp.reconstruct()

# 测试代码
data = np.random.randn(1024)
wavelet = 'db4'
level = 3

wp = wavelet_packet_decomposition(data, wavelet, level)
print(wp.shape)

reconstructed_data = wavelet_packet_reconstruction(wp, wavelet, level)
print(np.allclose(data, reconstructed_data))

在这个示例中，我们使用了PyWavelets库来实现小波包分解和重构。小波包分解函数将输入数据分解为多个小波包系数，而小波包重构函数将小波包系数重新组合成原始数据。在这个示例中，我们生成了一个随机数据向量，使用db4小波，并将小波包分解级别设置为3。我们将分解后的小波包系数打印出来，并使用小波包重构函数将它们重新组合成原始数据，并检查它们是否与原始数据相等。