python - matplotlib 颜色条放置和大小

Question

我正在使用 quadmesh 创建一个简单的极坐标投影图。这是一个最小的脚本，它基本上可以生成我正在尝试做的事情:

from __future__ import unicode_literals
import numpy as np
import matplotlib.pyplot as plt

def make_plot(data,fig,subplot):
    nphi,nt = data.shape
    phi_coords = np.linspace(0,np.pi*2,nphi+1) - np.pi/2.
    theta_coords = np.linspace(0,np.radians(35),nt+1)

    ax = fig.add_subplot(subplot,projection='polar')
    ax.set_thetagrids((45,90,135,180,225,270,315,360),(9,12,15,18,21,24,3,6))
    ax.set_rgrids(np.arange(10,35,10),fmt='%s\u00b0')  

    theta,phi = np.meshgrid(phi_coords,theta_coords)
    quadmesh = ax.pcolormesh(theta,phi,data)
    ax.grid(True)
    fig.colorbar(quadmesh,ax=ax)
    return fig,ax


a = np.zeros((360,71)) + np.arange(360)[:,None]
b = np.random.random((360,71))
fig = plt.figure()
t1 = make_plot(a,fig,121)
t2 = make_plot(b,fig,122)
fig.savefig('test.png')

上面的脚本创建了一个如下所示的图:

我希望颜色条:

1. 不与 6 标签重叠。
2. 进行缩放，使它们与地 block 的高度大致相同。

有什么技巧可以使它正常工作吗？ (请注意，这种布局不是我将使用的唯一布局——例如，我可能会使用 1x2 布局或 4x4 布局……似乎应该有某种方法可以将颜色栏缩放到与相关情节...)

Answer 1

这种组合(以及接近这些的值)似乎对我来说“神奇地”起作用，无论显示大小如何，都可以使颜色条按比例缩放到绘图中。

plt.colorbar(im,fraction=0.046, pad=0.04)