小波包分解特征提取python代码

以下是一个小波包分解特征提取的 Python 代码示例：

import pywt
import numpy as np

def wavelet_packet_decomposition(signal, wavelet='db1', level=4):
    wp = pywt.WaveletPacket(signal, wavelet, mode='symmetric', maxlevel=level)
    nodes = wp.get_level(level, 'natural')
    features = []
    for node in nodes:
        if node.path.startswith('d'):
            features.append(np.std(node.data))
        else:
            features.append(np.mean(node.data))
    return features

这个函数接受一个信号作为输入，使用小波包分解将其分解为多个子带，然后提取每个子带的标准差或平均值作为特征。这些特征可以用于分类、回归等任务。