将Colorbar添加到频谱图

Question

我正在尝试将色条添加到光谱图中.我已经尝试了我在网上找到的每一个例子和问题线程,但没有人解决过这个问题

请注意,'spl1'(数据拼接1)是来自ObsPy的跟踪.

我的代码是:

fig = plt.figure()
ax1 = fig.add_axes([0.1, 0.75, 0.7, 0.2]) #[left bottom width height]
ax2 = fig.add_axes([0.1, 0.1, 0.7, 0.60], sharex=ax1)
ax3 = fig.add_axes([0.83, 0.1, 0.03, 0.6])

t = np.arange(spl1[0].stats.npts) / spl1[0].stats.sampling_rate
ax1.plot(t, spl1[0].data, 'k')

ax,spec = spectrogram(spl1[0].data,spl1[0].stats.sampling_rate, show=False, axes=ax2)
ax2.set_ylim(0.1, 15)
fig.colorbar(spec, cax=ax3)

它出现了错误:

Traceback (most recent call last):

  File "<ipython-input-18-61226ccd2d85>", line 14, in <module>
    ax,spec = spectrogram(spl1[0].data,spl1[0].stats.sampling_rate, show=False, axes=ax2)

TypeError: 'Axes' object is not iterable

迄今为止的最佳结果:

用以下代码替换上面的最后3行:

ax = spectrogram(spl1[0].data,spl1[0].stats.sampling_rate, show=False, axes=ax2)
ax2.set_ylim(0.1, 15)
fig.colorbar(ax,cax=ax3)

产生这个:

和colorbar的这个错误:

axes object has no attribute 'autoscale_None'

我似乎无法找到一种方法来让右边的颜色条工作.

解决方案？

我见过的解决方案之一是你需要使用imshow()创建数据的"图像",但是我没有从Spectrogram()获得输出,只有'ax'.我已经看到地方尝试使用spectrogram()的'ax,spec'输出,但这会导致TypeError.

• 我找到的代码非常相似但无法正常工作https://www.nicotrebbin.de/wp-content/uploads/2012/03/bachelorthesis.pdf(ctrl + f'colorbar')
• 查看相关问题的代码示例
• imshow()建议和示例 - 无法从频谱图获得输出变成图像.第二个链接,我也无法使mlpy模块工作(它不认为有一个mlpy.wavelet函数)
• 这个问题在obspy的改进帖子中得到了解决,但他说他发现的解决方案没有给出

我希望有人可以帮忙 - 我一整天都在努力!

Answer 1

在此链接的帮助下解决了它。它尚未显示分贝，但主要问题是获取颜色条：

from obspy.imaging.spectrogram import spectrogram
fig = plt.figure()
ax1 = fig.add_axes([0.1, 0.75, 0.7, 0.2]) #[left bottom width height]
ax2 = fig.add_axes([0.1, 0.1, 0.7, 0.60], sharex=ax1)
ax3 = fig.add_axes([0.83, 0.1, 0.03, 0.6])

#make time vector
t = np.arange(spl1[0].stats.npts) / spl1[0].stats.sampling_rate

#plot waveform (top subfigure)    
ax1.plot(t, spl1[0].data, 'k')

#plot spectrogram (bottom subfigure)
spl2 = spl1[0]
fig = spl2.spectrogram(show=False, axes=ax2)
mappable = ax2.images[0]
plt.colorbar(mappable=mappable, cax=ax3)