对matplotlib中的不同子图使用相同的颜色条

时间:2016-12-05 18:30:10

标签: python matplotlib subplot colorbar

我正在使用以下程序在子图中绘制不同的数字。

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

但是我想为所有子图设置相同的颜色条,对于图的右侧的istance。

1 个答案:

答案 0 :(得分:2)

请看下面的示例:

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