python小波变换

def extract_wavelet_features(data, column_name='声波强度 (AE)', scales=np.arange(1, 4), wavelet='cmor'):

    emr_values = data[column_name].values

    coefficients, frequencies = pywt.cwt(emr_values, scales, wavelet)

    scale_energies = np.sum(np.abs(coefficients)**2, axis=0)

    scale_means = np.mean(np.abs(coefficients), axis=0)

    scale_stds = np.std(np.abs(coefficients), axis=0)