给出一段用python编写的小波软阈值函数

以下是一个简单的小波软阈值函数的Python实现：

import numpy as np
import pywt

def wavelet_soft_threshold(data, threshold):
    # 将数据进行小波变换
    coeffs = pywt.wavedec(data, 'db4', level=2)
    # 对每个系数进行软阈值处理
    for i in range(1, len(coeffs)):
        coeffs[i] = pywt.threshold(coeffs[i], threshold)
    # 将处理后的系数进行小波反变换
    data_rec = pywt.waverec(coeffs, 'db4')
    return data_rec

其中，`data`是输入的数据，`threshold`是软阈值的阈值。该函数使用pywt库进行小波变换和小波反变换，对每个小波系数进行软阈值处理，返回处理后的数据。