给出小波包分解的python代码

以下是小波包分解的Python代码：

import pywt

def wavelet_packet_decomposition(signal, waveletname, level):
    wp = pywt.WaveletPacket(signal, waveletname, mode='symmetric', maxlevel=level)
    
    # 获取每个节点的能量
    nodes_energy = {}
    for node in wp.get_level(level, 'natural'):
        if not node.is_leaf:
            continue
        nodes_energy[node.path] = node.data ** 2

    # 根据能量排序，选出前 K 个保留
    K = 8  # 设置保留的节点数
    sorted_nodes = sorted(nodes_energy.items(), key=lambda x: x[1], reverse=True)[:K]

    # 重新构建小波分解结果
    nodes_to_keep = set(node_path for node_path, _ in sorted_nodes)
    for node in wp.get_level(level, 'natural'):
        if node.path not in nodes_to_keep:
            node.data = None
    return wp.reconstruct(update=True)

希望这个代码可以对你有帮助。