python - 在Matplotlib中使用子图时如何控制颜色条的位置

This question already has an answer here:

adjusting subplot with a colorbar
                                
                                    （1个答案）
                                
                        
                                在11个月前关闭。
            
                    
我正在尝试使用Matplotlib绘制时间序列及其频谱图和关联的颜色条。

以下是MCVE：

import numpy as np
import matplotlib.pyplot as plt
import scipy.signal as scignal
import random

array=np.random.random(10000)
t,f,Sxx=scignal.spectrogram(array,fs=100)

plt.subplot(211)
plt.plot(array)
plt.subplot(212)
plt.pcolormesh(Sxx)
plt.colorbar()

此代码产生下图：



但是，我希望两个子图具有相同的大小：



我曾想过使用plt.colorbar(orientation='horizontal')更改颜色条的方向，但由于子图的高度不同，我对结果不满意。

任何帮助将不胜感激！

最佳答案

发生这种情况的原因是plt.colorbar创建了一个新的Axes对象，该对象从较低的Axes“窃取”空间（这是制作水平颜色条也会影响两个原始图的原因）。

有几种方法可以解决此问题。一种是用四个Figure创建一个Axes，将大部分空间分配给左边的空间，然后使其中一个不可见：

import numpy as np
import matplotlib.pyplot as plt
import scipy.signal as scignal
import random

array = np.random.random(10000)
t, f, Sxx = scignal.spectrogram(array,fs=100)

fig, axes = plt.subplots(nrows=2, ncols=2, figsize=(5, 6), gridspec_kw={'width_ratios': [19, 1]})
(ax1, blank), (ax2, ax_cb) = axes
blank.set_visible(False)
ax1.plot(array)
m = ax2.pcolormesh(Sxx)
fig.colorbar(m, cax=ax_cb)

关于python - 在Matplotlib中使用子图时如何控制颜色条的位置，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/55607536/