如何通过频谱图绘制时频图？

Question

我正试着通过频谱图绘制时频图。我想做一个线图(x轴=时间，y轴=频率)。我使用了Python和signal.stft函数，并得到了一个谱图。我还计算了每个时间步的最大频率值。但是我得不到答案。有什么问题吗？

Answer 1

如果我没理解错的话，你想在频谱图上画一条最大频率的曲线？如果是，您可以简单地在频谱图之后调用matplotlib的第二个plot调用。如下所示：

f, t, Zxx = signal.stft(x, fs, nperseg=1000)
max_f = np.argmax(data, axis=1)  # data has shape (time x frequency)
plt.pcolormesh(t, f, np.abs(Zxx), vmin=0, vmax=amp)
plt.plot(np.arange(max_f.shape[0]), max_f)
plt.ylabel('Frequency [Hz]')
plt.xlabel('Time [sec]')
plt.show()