Use the same colorbar for different subplots in matplotlib

emax picture emax · Dec 5, 2016 · Viewed 7.3k times · Source

I am plotting different figure in subplots using the following procedure.

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

and this is what I get:

enter image description here

However I would like to have the same colorbar for all the subplots, for istance on the right side of the figure.

Answer

Serenity picture Serenity · Dec 5, 2016

Look at the example below:

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