pywt 自定义小波（python）

如下，是自定义的db2小波

import pywt, math
import numpy as np
# c = math.sqrt(2)/2
# dec_lo, dec_hi, rec_lo, rec_hi = [c, c], [-c, c], [c, c], [c, -c]
# filter_bank = [dec_lo, dec_hi, rec_lo, rec_hi]
# myWavelet = pywt.Wavelet(name="myHaarWavelet", filter_bank=filter_bank)

class HaarFilterBank(object):
    @property
    def filter_bank(self):
        s3 = math.sqrt(3)
        d = 4 * math.sqrt(2)

        h0 = (1 + s3) / d
        h1 = (3 + s3) / d
        h2 = (3 - s3) / d
        h3 = (1 - s3) / d

        g0 = h3
        g1 = -h2
        g2 = h1
        g3 = -h0
        # c = math.sqrt(2)/2
        dec_lo, dec_hi, rec_lo, rec_hi = [h3,h2,h1,h0], [g3,g2,g1,g0], [h0,h1,h2,h3], [g0,g1,g2,g3]
        return [dec_lo, dec_hi, rec_lo, rec_hi]

filter_bank = HaarFilterBank()
myOtherWavelet = pywt.Wavelet(name="myHaarWavelet", filter_bank=filter_bank)

# print(pywt.Wavelet("myHaarWavelet"))
print(myOtherWavelet)
print(pywt.Wavelet("db1"))
A = np.array([1,2,3,4,5,6])
B = pywt.dwt(A,"db1")
print(B)
B = pywt.idwt(B[0],B[1],'db1')
print(B)

B = pywt.dwt(A,myOtherWavelet)
print(B)
B = pywt.idwt(B[0],B[1],myOtherWavelet)
print(B)