python中cwt函数用法

在Python中，cwt函数是用于进行连续小波变换（CWT）的函数。它的用法如下：

cwt(data, wavelet, widths)

其中，参数说明如下：

- data：要进行CWT的数据，可以是一维或二维数组。
- wavelet：小波函数，可以使用pywt包中提供的小波函数，例如“morl”，“cgau5”等。
- widths：小波尺度，可以是一个数字，表示要进行多少次小波变换，也可以是一个数组，表示要进行不同尺度的小波变换。

该函数的返回值是一个二维数组，表示在不同尺度和时间上的小波系数。

示例代码：

import numpy as np
import pywt

# 生成测试数据
t = np.linspace(-1, 1, 200, endpoint=False)
sig = np.sin(2 * np.pi * 7 * t) + np.cos(2 * np.pi * 2 * t) + np.sin(2 * np.pi * 14 * t)

# 进行CWT
widths = np.arange(1, 31)
cwtmatr, freqs = pywt.cwt(sig, widths, 'morl')

# 绘制结果
import matplotlib.pyplot as plt

plt.figure(figsize=(6, 4))
plt.imshow(abs(cwtmatr), extent=[-1, 1, 1, 31], cmap='gray', aspect='auto',
           vmax=abs(cwtmatr).max(), vmin=-abs(cwtmatr).max())
plt.xlabel('Time (s)')
plt.ylabel('Scale')
plt.show()

该示例代码生成一个测试信号，然后使用“morl”小波函数和30个不同尺度进行CWT，并将结果绘制成图像。运行结果如下所示：