python小波变换_小波变换学习-pywavelets

最近正在学习音频方面的项目，涉及到了小波变换的内容。pywavelets是其中一个比较常见的小波模块。复现了下其他博客如下：小波变换-python pywavelets

np.piecewise是根据条件筛选，然后执行例如lambda/函数等操作。参考链接np.piecewise

#!/usr/bin/env python

# encoding: utf-8

import numpy as np

import matplotlib.pyplot as plt

import pywt

sampling_rate = 1024#采样频率

t = np.arange(0,1.0,1.0/sampling_rate)

f1 = 100#频率

f2 = 200

f3 = 300

data = np.piecewise(t,[t<1,t<0.8,t<0.3],

[lambda t : np.sin(2*np.pi *f1*t),

lambda t : np.sin(2 * np.pi * f2 * t),

lambda t : np.sin(2 * np.pi * f3 * t)])

wavename = "cgau8"

totalscal = 256

fc = pywt.central_frequency(wavename)#中心频率

cparam = 2 * fc * totalscal

scales = cparam/np.arange(totalscal,1,-1)

[cwtmatr, frequencies] = pywt.cwt(data,scales,wavename,1.0/sampling_rate)#连续小波变换

plt.figure(f