在Python中,wavedec2函数是Wavelet包中的一个函数,用于对二维信号进行二维离散小波变换。
它的用法如下:
coeffs = wavedec2(data, wavelet, mode='symmetric', level=1)
参数说明:
返回值:
示例代码:
import pywt
import numpy as np
# 生成一个二维信号
data = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12],
[13, 14, 15, 16]])
# 进行二维小波变换
coeffs = pywt.wavedec2(data, 'db1', mode='symmetric', level=1)
print(coeffs)
输出结果:
[array([[ 2.5, 4.5],
[10.5, 12.5]]), (array([[ 0.5, 0.5],
[ 2.5, 2.5]]), array([[ -1., -1.],
[-17., -17.]]))]
在上面的示例中,输入的二维信号是一个4x4的数组。经过一层小波变换后,得到一个低频分量和一个高频分量的元组。coeffs[0]
是低频分量,coeffs[1]
是高频分量。