在Python中matplotlib不均勻居中的subplots？

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()) 

我得到的情節附後。頂部和底部的地塊不均勻居中。我希望他們的yaxis被對齊並且他們的盒子被對齊。

如果我刪除了plt.colorbar（）調用，那麼這些圖將變爲居中。當pcolor的顏色條仍然顯示時，我怎樣才能使圖形居中？我想讓座標軸居中並使顏色條位於該對齊線之外，或者位於pcolor矩陣的左側或右側。

image of plots link

Image of uncentered subplots in matplotlib http://i45.tinypic.com/2i1kvtu.png

感謝。

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

給一個cax參數彩條，指定其中你喜歡的顏色條：

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