在matplotlib中,为不同的子图使用相同的色条

7
我正在使用以下步骤在子图中绘制不同的图表。
fig = figure(figsize=(10,11))
subplots_adjust(wspace=0.5,hspace=0.2)
iplot = 330
for i in range(9):
    iplot += 1
    ax = fig.add_subplot(iplot)
    ## Comparison Mreal - Real
    tmp = REAL[REAL.days==days[i]]
    tmp = tmp.score
    tmp = np.array(tmp)
    tmp = tmp.reshape(len(xv), len(br))
    im = plt.imshow(tmp, interpolation='nearest', cmap='gnuplot', vmin = 0, vmax = 1,  extent=[0.05,0.5,1,0.05],
              aspect=0.5)
    xtmp = [0.05, 0.2, 0.3, 0.4, 0.5]
    plt.xticks(xtmp)
    ytmp = [0.05, 0.2, 0.4, 0.6, 0.8, 1.0]
    plt.yticks(ytmp)
    ax.grid(False)
divider = make_axes_locatable(plt.gca())
cax = divider.append_axes("right", "5%", pad="3%")
plt.colorbar(im, cax=cax)
plt.tight_layout()

这是我得到的结果:

enter image description here

然而,我希望所有子图都有相同的色条,例如在图的右侧。
1个回答

7

看下面的例子:

import matplotlib.pyplot as plt

fig, axes = plt.subplots(nrows=3, ncols=3)
for ax in axes.flat:
    im = ax.imshow(np.random.random((6,6)), interpolation='nearest', cmap='gnuplot',
     vmin=0, vmax=1, extent=[0.05,0.5,1,0.05],aspect=0.5)

fig.subplots_adjust(right=0.8)
# put colorbar at desire position
cbar_ax = fig.add_axes([0.85, 0.15, 0.05, 0.7])
fig.colorbar(im, cax=cbar_ax)

plt.show()

enter image description here


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接