演示Axes Grid2

共享xaxis和yaxis的图像网格。

演示Axes Grid2

  1. import matplotlib.pyplot as plt
  2. from mpl_toolkits.axes_grid1 import ImageGrid
  3. import numpy as np
  6. def get_demo_image():
  7.     from matplotlib.cbook import get_sample_data
  8.     f = get_sample_data("axes_grid/bivariate_normal.npy", asfileobj=False)
  9.     z = np.load(f)
  10.     # z is a numpy array of 15x15
  11.     return z, (-3, 4, -4, 3)
  14. def add_inner_title(ax, title, loc, size=None, **kwargs):
  15.     from matplotlib.offsetbox import AnchoredText
  16.     from matplotlib.patheffects import withStroke
  17.     if size is None:
  18.         size = dict(size=plt.rcParams['legend.fontsize'])
  19.     at = AnchoredText(title, loc=loc, prop=size,
  20.                       pad=0., borderpad=0.5,
  21.                       frameon=False, **kwargs)
  22.     ax.add_artist(at)
  23.     at.txt._text.set_path_effects([withStroke(foreground="w", linewidth=3)])
  24.     return at
  26. if 1:
  27.     F = plt.figure(1, (6, 6))
  28.     F.clf()
  30.     # prepare images
  31.     Z, extent = get_demo_image()
  32.     ZS = [Z[i::3, :] for i in range(3)]
  33.     extent = extent[0], extent[1]/3., extent[2], extent[3]
  35.     # demo 1 : colorbar at each axes
  37.     grid = ImageGrid(F, 211,  # similar to subplot(111)
  38.                      nrows_ncols=(1, 3),
  39.                      direction="row",
  40.                      axes_pad=0.05,
  41.                      add_all=True,
  42.                      label_mode="1",
  43.                      share_all=True,
  44.                      cbar_location="top",
  45.                      cbar_mode="each",
  46.                      cbar_size="7%",
  47.                      cbar_pad="1%",
  48.                      )
  50.     for ax, z in zip(grid, ZS):
  51.         im = ax.imshow(
  52.             z, origin="lower", extent=extent, interpolation="nearest")
  53.         ax.cax.colorbar(im)
  55.     for ax, im_title in zip(grid, ["Image 1", "Image 2", "Image 3"]):
  56.         t = add_inner_title(ax, im_title, loc='lower left')
  57.         t.patch.set_alpha(0.5)
  59.     for ax, z in zip(grid, ZS):
  60.         ax.cax.toggle_label(True)
  61.         #axis = ax.cax.axis[ax.cax.orientation]
  62.         #axis.label.set_text("counts s$^{-1}$")
  63.         #axis.label.set_size(10)
  64.         #axis.major_ticklabels.set_size(6)
  66.     # changing the colorbar ticks
  67.     grid[1].cax.set_xticks([-1, 0, 1])
  68.     grid[2].cax.set_xticks([-1, 0, 1])
  70.     grid[0].set_xticks([-2, 0])
  71.     grid[0].set_yticks([-2, 0, 2])
  73.     # demo 2 : shared colorbar
  75.     grid2 = ImageGrid(F, 212,
  76.                       nrows_ncols=(1, 3),
  77.                       direction="row",
  78.                       axes_pad=0.05,
  79.                       add_all=True,
  80.                       label_mode="1",
  81.                       share_all=True,
  82.                       cbar_location="right",
  83.                       cbar_mode="single",
  84.                       cbar_size="10%",
  85.                       cbar_pad=0.05,
  86.                       )
  88.     grid2[0].set_xlabel("X")
  89.     grid2[0].set_ylabel("Y")
  91.     vmax, vmin = np.max(ZS), np.min(ZS)
  92.     import matplotlib.colors
  93.     norm = matplotlib.colors.Normalize(vmax=vmax, vmin=vmin)
  95.     for ax, z in zip(grid2, ZS):
  96.         im = ax.imshow(z, norm=norm,
  97.                        origin="lower", extent=extent,
  98.                        interpolation="nearest")
  100.     # With cbar_mode="single", cax attribute of all axes are identical.
  101.     ax.cax.colorbar(im)
  102.     ax.cax.toggle_label(True)
  104.     for ax, im_title in zip(grid2, ["(a)", "(b)", "(c)"]):
  105.         t = add_inner_title(ax, im_title, loc='upper left')
  106.         t.patch.set_ec("none")
  107.         t.patch.set_alpha(0.5)
  109.     grid2[0].set_xticks([-2, 0])
  110.     grid2[0].set_yticks([-2, 0, 2])
  112.     plt.show()

下载这个示例