使用matplotlib中的specgram创建对数频率轴频谱图

时间:2012-05-30 07:55:11

标签: python matplotlib scipy

我想知道这个功能是否已通过specgram内置,我相信情况并非如此。如果没有,那么使用specgram或matplotlib实现此方法的最佳方法。如果specgram轴的某些变换可以启用此功能。

例如,对于specgram图,下面的代码不起作用(任何音频信号都可以作为specgram的第一个参数传递):

fig = figure(11)
ax = fig.add_subplot(111)
ax.specgram(tidig.train_symbol_list[0].audio,Fs=12000)
ax.set_yscale('log')

1 个答案:

答案 0 :(得分:1)

您可以使用Axes.set_xscaleAxes.set_yscale函数更改轴的缩放比例,这两个函数都接受linearlogsymlog作为输入。因此,要将绘图更改为具有对数缩放的x轴,您可以执行以下操作:

import matplotlib.pyplot as plt

ha = plt.subplot(111)

# Plot your spectrogram here...

ha.set_xscale('log')

修改这似乎是known issue。有一些命令可以执行此操作,但不是以任何特别方便的方式(它应该只是specgram函数的标志或set_xscaleset_yscale应该工作)。但是,有一种方法可以做到这一点:

而不是使用matplotlib.pyplot.specgram使用matplotlib.mlab.specgram。这会计算出频谱,但不会绘制它。然后,您可以使用matplotlib.pyplot.pcolor或类似功能绘制视图。所以尝试类似的事情:

import matplotlib.pyplot as plt
import matplotlib.mlab as mlab

# Set up your data here...

Pxx, freq, t = mlab.specgram(x) # Other options, such as NFFT, may be passed
                                # to `specgram` here.

ha = plt.subplot(111)
ha.pcolor(t, freq, Pxx)
ha.set_xscale('log')