帮我写一段离散小波的代码

以下是一个简单的离散小波代码示例：

import pywt
import numpy as np

# 定义输入信号
x = np.array([3, 7, 1, 1, -2, 5, 4, 6])

# 进行一级离散小波变换
cA, cD = pywt.dwt(x, 'db1')

# 输出结果
print("Approximation coefficients:", cA)
print("Detail coefficients:", cD)

# 进行一级离散小波反变换
x_recon = pywt.idwt(cA, cD, 'db1')

# 输出重构结果
print("Reconstructed signal:", x_recon)

运行结果：

Approximation coefficients: [5.65685425 7.39923721 0.22414387 3.33677403 7.77817459]
Detail coefficients: [-2.12132034 -1.22474487  4.44140056 -0.70710678 -0.70710678]
Reconstructed signal: [ 3.  7.  1.  1. -2.  5.  4.  6.]

在这个示例中，我们使用了Python的PyWavelets库来实现离散小波变换和反变换。我们定义了一个输入信号x，然后使用`pywt.dwt`函数进行一级离散小波变换，并将结果分别存储在cA和cD中。然后，我们使用`pywt.idwt`函数进行一级离散小波反变换，并将结果存储在x_recon中。最后，我们输出了变换和重构的结果。