如何解决AttributeError: module 'pywt' has no attribute 'sure_thresh'？

错误原因 #

你遇到的AttributeError是因为PyWavelets（pywt）库中不存在threshold_sure或sure_thresh这类函数。SureShrink方法的实现需要分两步：先通过pywt.sure_threshold计算基于SURE准则的阈值，再用pywt.threshold对小波系数应用阈值。

正确实现代码 #

以下是完整的SureShrink小波阈值降噪示例，包含小波分解、阈值计算、系数处理和信号重构的全流程：

import numpy as np
import pywt

# 示例时间序列（替换为你的实际数据）
np.random.seed(42)
raw_data = np.sin(np.linspace(0, 10, 1000)) + 0.2 * np.random.randn(1000)

# 1. 小波分解（选择db4小波，分解3层，可根据需求调整）
wavelet_type = 'db4'
decomp_level = 3
coeffs = pywt.wavedec(raw_data, wavelet_type, level=decomp_level)

# 2. 对各层细节系数应用SureShrink阈值处理
for i in range(1, len(coeffs)):
    # 计算SureShrink阈值
    sure_threshold_val = pywt.sure_threshold(coeffs[i], mode='soft')
    # 应用软阈值（SureShrink默认使用软阈值）
    coeffs[i] = pywt.threshold(coeffs[i], sure_threshold_val, mode='soft')

# 3. 重构降噪后的信号
denoised_data = pywt.waverec(coeffs, wavelet_type)

关键说明 #

• pywt.sure_threshold：专门用于计算SureShrink方法的阈值，传入对应层的小波细节系数即可返回阈值，mode='soft'对应SureShrink的软阈值逻辑。
• pywt.threshold：负责将计算好的阈值应用到系数上，完成降噪的核心处理。
• 版本兼容：如果你的PyWavelets版本过旧，可能没有sure_threshold函数，执行pip install --upgrade pywavelets升级到最新版本即可解决。

内容的提问来源于stack exchange，提问作者chengguan86