Python中的matplotlib中的子图中心不均匀？

Question

我正在 matplotlib 中绘制一对简单的子图，由于某种原因，它们的中心不均匀。我将它们绘制如下：

plt.figure()
# first subplot
s1 = plt.subplot(2, 1, 1)
plt.bar([1, 2, 3], [4, 5, 6])
# second subplot
s2 = plt.subplot(2, 1, 2)
plt.pcolor(rand(5,5))

# add colorbar
plt.colorbar()

# square axes
axes_square(s1)
axes_square(s2)

其中axes_square很简单：

def axes_square(plot_handle):
  plot_handle.axes.set_aspect(1/plot_handle.axes.get_data_ratio()) 

我得到的图已附上。顶部和底部图的中心不均匀。我希望它们的 y 轴对齐并且它们的框对齐。

如果我删除 plt.colorbar() 调用，绘图就会居中。如何在仍显示 pcolor 的颜色条的情况下使绘图居中？我希望轴居中，并且颜色条位于该对齐方式之外，位于 pcolor 矩阵的左侧或右侧。

绘图链接图像

谢谢。

Answer 1

好吧，这可能不完全是您想要的，但它会起作用：

from numpy.random import rand
import matplotlib.pyplot as plt

plt.figure()
# first subplot
s1 = plt.subplot(2, 2, 2)
plt.bar([1, 2, 3], [4, 5, 6])
# second subplot
s2 = plt.subplot(2, 2, 4)
plt.pcolor(rand(5,5))

# square axes
axes_square(s1)
axes_square(s2)

s_fake = plt.subplot(2, 2, 3)
s_fake.set_axis_off()
# add colorbar
plt.colorbar()

它只是在左侧生成一对假单元格，并且不会在其中显示任何内容。不漂亮，但它有效。 ;)

另外，我猜测这两个导入是隐含在您的代码中的......

Answer 2

为 colorbar 提供一个 cax 参数来指定您喜欢的 colorbar 位置：

plt.colorbar(cax=plt.gcf().add_axes((0.75,0.1,0.05,0.3)))