演示嵌入轴颜色条

演示嵌入轴颜色条

  1. import matplotlib.pyplot as plt
  3. from mpl_toolkits.axes_grid1.inset_locator import inset_axes, zoomed_inset_axes
  4. from mpl_toolkits.axes_grid1.colorbar import colorbar
  7. def get_demo_image():
  8.     from matplotlib.cbook import get_sample_data
  9.     import numpy as np
  10.     f = get_sample_data("axes_grid/bivariate_normal.npy", asfileobj=False)
  11.     z = np.load(f)
  12.     # z is a numpy array of 15x15
  13.     return z, (-3, 4, -4, 3)
  16. fig, ax = plt.subplots(figsize=[5, 4])
  18. Z, extent = get_demo_image()
  20. ax.set(aspect=1,
  21.        xlim=(-15, 15),
  22.        ylim=(-20, 5))
  25. axins = zoomed_inset_axes(ax, zoom=2, loc='upper left')
  26. im = axins.imshow(Z, extent=extent, interpolation="nearest",
  27.                   origin="lower")
  29. plt.xticks(visible=False)
  30. plt.yticks(visible=False)
  33. # colorbar
  34. cax = inset_axes(axins,
  35.                  width="5%",  # width = 10% of parent_bbox width
  36.                  height="100%",  # height : 50%
  37.                  loc='lower left',
  38.                  bbox_to_anchor=(1.05, 0., 1, 1),
  39.                  bbox_transform=axins.transAxes,
  40.                  borderpad=0,
  41.                  )
  43. colorbar(im, cax=cax)
  45. plt.show()

下载这个示例