演示Axes Grid

具有单个或自己的彩条的2x2图像的网格。

演示Axes Grid

  1. import matplotlib.pyplot as plt
  2. from mpl_toolkits.axes_grid1 import ImageGrid
  5. def get_demo_image():
  6.     import numpy as np
  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 demo_simple_grid(fig):
  15.     """
  16.     A grid of 2x2 images with 0.05 inch pad between images and only
  17.     the lower-left axes is labeled.
  18.     """
  19.     grid = ImageGrid(fig, 141,  # similar to subplot(141)
  20.                      nrows_ncols=(2, 2),
  21.                      axes_pad=0.05,
  22.                      label_mode="1",
  23.                      )
  25.     Z, extent = get_demo_image()
  26.     for i in range(4):
  27.         im = grid[i].imshow(Z, extent=extent, interpolation="nearest")
  29.     # This only affects axes in first column and second row as share_all =
  30.     # False.
  31.     grid.axes_llc.set_xticks([-2, 0, 2])
  32.     grid.axes_llc.set_yticks([-2, 0, 2])
  35. def demo_grid_with_single_cbar(fig):
  36.     """
  37.     A grid of 2x2 images with a single colorbar
  38.     """
  39.     grid = ImageGrid(fig, 142,  # similar to subplot(142)
  40.                      nrows_ncols=(2, 2),
  41.                      axes_pad=0.0,
  42.                      share_all=True,
  43.                      label_mode="L",
  44.                      cbar_location="top",
  45.                      cbar_mode="single",
  46.                      )
  48.     Z, extent = get_demo_image()
  49.     for i in range(4):
  50.         im = grid[i].imshow(Z, extent=extent, interpolation="nearest")
  51.     grid.cbar_axes[0].colorbar(im)
  53.     for cax in grid.cbar_axes:
  54.         cax.toggle_label(False)
  56.     # This affects all axes as share_all = True.
  57.     grid.axes_llc.set_xticks([-2, 0, 2])
  58.     grid.axes_llc.set_yticks([-2, 0, 2])
  61. def demo_grid_with_each_cbar(fig):
  62.     """
  63.     A grid of 2x2 images. Each image has its own colorbar.
  64.     """
  66.     grid = ImageGrid(fig, 143,  # similar to subplot(143)
  67.                      nrows_ncols=(2, 2),
  68.                      axes_pad=0.1,
  69.                      label_mode="1",
  70.                      share_all=True,
  71.                      cbar_location="top",
  72.                      cbar_mode="each",
  73.                      cbar_size="7%",
  74.                      cbar_pad="2%",
  75.                      )
  76.     Z, extent = get_demo_image()
  77.     for i in range(4):
  78.         im = grid[i].imshow(Z, extent=extent, interpolation="nearest")
  79.         grid.cbar_axes[i].colorbar(im)
  81.     for cax in grid.cbar_axes:
  82.         cax.toggle_label(False)
  84.     # This affects all axes because we set share_all = True.
  85.     grid.axes_llc.set_xticks([-2, 0, 2])
  86.     grid.axes_llc.set_yticks([-2, 0, 2])
  89. def demo_grid_with_each_cbar_labelled(fig):
  90.     """
  91.     A grid of 2x2 images. Each image has its own colorbar.
  92.     """
  94.     grid = ImageGrid(fig, 144,  # similar to subplot(144)
  95.                      nrows_ncols=(2, 2),
  96.                      axes_pad=(0.45, 0.15),
  97.                      label_mode="1",
  98.                      share_all=True,
  99.                      cbar_location="right",
  100.                      cbar_mode="each",
  101.                      cbar_size="7%",
  102.                      cbar_pad="2%",
  103.                      )
  104.     Z, extent = get_demo_image()
  106.     # Use a different colorbar range every time
  107.     limits = ((0, 1), (-2, 2), (-1.7, 1.4), (-1.5, 1))
  108.     for i in range(4):
  109.         im = grid[i].imshow(Z, extent=extent, interpolation="nearest",
  110.                             vmin=limits[i][0], vmax=limits[i][1])
  111.         grid.cbar_axes[i].colorbar(im)
  113.     for i, cax in enumerate(grid.cbar_axes):
  114.         cax.set_yticks((limits[i][0], limits[i][1]))
  116.     # This affects all axes because we set share_all = True.
  117.     grid.axes_llc.set_xticks([-2, 0, 2])
  118.     grid.axes_llc.set_yticks([-2, 0, 2])
  121. if 1:
  122.     F = plt.figure(1, (10.5, 2.5))
  124.     F.subplots_adjust(left=0.05, right=0.95)
  126.     demo_simple_grid(F)
  127.     demo_grid_with_single_cbar(F)
  128.     demo_grid_with_each_cbar(F)
  129.     demo_grid_with_each_cbar_labelled(F)
  131.     plt.show()

下载这个示例