Python环境下小波函数的绘制

摘要：import itertoolsimport matplotlib.pyplot as pltimport pywtplot_data = [('db', (4, 3)),('sym', (4, 3)),('coif', (3, 2))]for family,

import itertoolsimport matplotlib.pyplot as pltimport pywtplot_data = [('db', (4, 3)),('sym', (4, 3)),('coif', (3, 2))]for family, (rows, cols) in plot_data:fig = plt.figurefig.subplots_adjust(hspace=0.2, wspace=0.2, bottom=.02, left=.06,right=.97, top=.94)colors = itertools.cycle('bgrcmyk')wnames = pywt.wavelist(family)i = iter(wnames)for col in range(cols):for row in range(rows):try:wavelet = pywt.Wavelet(next(i))except StopIteration:breakphi, psi, x = wavelet.wavefun(level=5)color = next(colors)ax = fig.add_subplot(rows, 2 * cols, 1 + 2 * (col + row * cols))ax.set_title(wavelet.name + " phi")ax.plot(x, phi, color)ax.set_xlim(min(x), max(x))ax = fig.add_subplot(rows, 2*cols, 1 + 2*(col + row*cols) + 1)ax.set_title(wavelet.name + " psi")ax.plot(x, psi, color)ax.set_xlim(min(x), max(x))for family, (rows, cols) in [('bior', (4, 3)), ('rbio', (4, 3))]:fig = plt.figurefig.subplots_adjust(hspace=0.5, wspace=0.2, bottom=.02, left=.06,right=.97, top=.94)colors = itertools.cycle('bgrcmyk')wnames = pywt.wavelist(family)i = iter(wnames)for col in range(cols):for row in range(rows):try:wavelet = pywt.Wavelet(next(i))except StopIteration:breakphi, psi, phi_r, psi_r, x = wavelet.wavefun(level=5)row *= 2color = next(colors)ax = fig.add_subplot(2*rows, 2*cols, 1 + 2*(col + row*cols))ax.set_title(wavelet.name + " phi")ax.plot(x, phi, color)ax.set_xlim(min(x), max(x))ax = fig.add_subplot(2*rows, 2*cols, 2*(1 + col + row*cols))ax.set_title(wavelet.name + " psi")ax.plot(x, psi, color)ax.set_xlim(min(x), max(x))row += 1ax = fig.add_subplot(2*rows, 2*cols, 1 + 2*(col + row*cols))ax.set_title(wavelet.name + " phi_r")ax.plot(x, phi_r, color)ax.set_xlim(min(x), max(x))ax = fig.add_subplot(2*rows, 2*cols, 1 + 2*(col + row*cols) + 1)ax.set_title(wavelet.name + " psi_r")ax.plot(x, psi_r, color)ax.set_xlim(min(x), max(x))plt.show