python编一段代码对语音信号进行STFT

以下是Python代码，用于对语音信号进行STFT：

# 导入所需的库
import numpy as np
import scipy.signal as sig
import matplotlib.pyplot as plt

# 定义一个语音信号
fs = 44100 # 采样率
t = np.linspace(0, 1, fs, endpoint=False) # 时间轴
f1 = 100 # 基频
f2 = 200 # 第二个频率
x = np.sin(2*np.pi*f1*t) + 0.5*np.sin(2*np.pi*f2*t) # 语音信号

# 进行STFT
window_size = int(0.02*fs) # 窗口大小
step_size = int(0.01*fs) # 步长
f, t, Zxx = sig.stft(x, fs=fs, window='hamming', nperseg=window_size, noverlap=window_size-step_size)

# 绘制STFT结果
plt.pcolormesh(t, f, np.abs(Zxx), cmap='jet')
plt.colorbar()
plt.title('STFT Magnitude')
plt.ylabel('Frequency [Hz]')
plt.xlabel('Time [sec]')
plt.show()

解释：

1. 首先，定义一个简单的语音信号，包含两个频率分别为100 Hz和200 Hz的正弦波。

2. 接下来，定义STFT所需的窗口大小和步长，并使用`scipy.signal.stft()`函数计算STFT。此函数的参数包括语音信号、采样率、窗口类型、窗口大小、重叠大小等。

3. 最后，使用`matplotlib`库绘制STFT的结果，其中横轴为时间，纵轴为频率，颜色表示幅度大小。