首页 > 解决方案 > 存储 pcolormesh 的 Python matplotlib 函数的输出并重建它

问题描述

我将对这个信号应用短时傅里叶变换,并使用pcolormesh函数绘制结果。然后,我想使用逆短时傅里叶变换来重建信号。我的问题是如何使用 pcolormesh 的输出作为 signal.istft(Zxx) 的输入来获取信号。简而言之,这就是我想要做的:

f, t, Zxx = signal.stft(sig, fs=250, nperseg=64)
mesh=plt.pcolormesh(t, f, np.abs(Zxx))
plt.axis('off')
plt.show()

输出此图像。然后,应用傅里叶逆变换重建信号:

_, xrec = signal.istft(mesh, fs=250)

实际上,我不确定上述功能的网格输入是否正确?以及如何获得 signal.istft(Zxx) 而不是 Zxx 的正确输入?

谢谢你的帮助。

标签: pythonmatplotlibscipyifft

解决方案


推荐阅读