插入定位器演示2

本演示展示了如何通过 zoomed_inset_axes 创建缩放插图。 在第一个子图中，AnchoredSizeBar 显示缩放效果。 在第二个子图中，通过mark_inset 创建与感兴趣区域的连接。

插入定位器演示2

  1. import matplotlib.pyplot as plt
  3. from mpl_toolkits.axes_grid1.inset_locator import zoomed_inset_axes, mark_inset
  4. from mpl_toolkits.axes_grid1.anchored_artists import AnchoredSizeBar
  6. import numpy as np
  9. def get_demo_image():
  10.     from matplotlib.cbook import get_sample_data
  11.     import numpy as np
  12.     f = get_sample_data("axes_grid/bivariate_normal.npy", asfileobj=False)
  13.     z = np.load(f)
  14.     # z is a numpy array of 15x15
  15.     return z, (-3, 4, -4, 3)
  17. fig, (ax, ax2) = plt.subplots(ncols=2, figsize=[6, 3])
  20. # First subplot, showing an inset with a size bar.
  21. ax.set_aspect(1)
  23. axins = zoomed_inset_axes(ax, zoom=0.5, loc='upper right')
  24. # fix the number of ticks on the inset axes
  25. axins.yaxis.get_major_locator().set_params(nbins=7)
  26. axins.xaxis.get_major_locator().set_params(nbins=7)
  28. plt.setp(axins.get_xticklabels(), visible=False)
  29. plt.setp(axins.get_yticklabels(), visible=False)
  32. def add_sizebar(ax, size):
  33.     asb = AnchoredSizeBar(ax.transData,
  34.                           size,
  35.                           str(size),
  36.                           loc=8,
  37.                           pad=0.1, borderpad=0.5, sep=5,
  38.                           frameon=False)
  39.     ax.add_artist(asb)
  41. add_sizebar(ax, 0.5)
  42. add_sizebar(axins, 0.5)
  45. # Second subplot, showing an image with an inset zoom
  46. # and a marked inset
  47. Z, extent = get_demo_image()
  48. Z2 = np.zeros([150, 150], dtype="d")
  49. ny, nx = Z.shape
  50. Z2[30:30 + ny, 30:30 + nx] = Z
  52. # extent = [-3, 4, -4, 3]
  53. ax2.imshow(Z2, extent=extent, interpolation="nearest",
  54.           origin="lower")
  57. axins2 = zoomed_inset_axes(ax2, 6, loc=1)  # zoom = 6
  58. axins2.imshow(Z2, extent=extent, interpolation="nearest",
  59.               origin="lower")
  61. # sub region of the original image
  62. x1, x2, y1, y2 = -1.5, -0.9, -2.5, -1.9
  63. axins2.set_xlim(x1, x2)
  64. axins2.set_ylim(y1, y2)
  65. # fix the number of ticks on the inset axes
  66. axins2.yaxis.get_major_locator().set_params(nbins=7)
  67. axins2.xaxis.get_major_locator().set_params(nbins=7)
  69. plt.setp(axins2.get_xticklabels(), visible=False)
  70. plt.setp(axins2.get_yticklabels(), visible=False)
  72. # draw a bbox of the region of the inset axes in the parent axes and
  73. # connecting lines between the bbox and the inset axes area
  74. mark_inset(ax2, axins2, loc1=2, loc2=4, fc="none", ec="0.5")
  76. plt.show()

下载这个示例