Viewlims

创建两个相同的面板。在右侧面板上放大将在第一个面板中显示一个矩形，表示缩放的区域。

Viewlims

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. from matplotlib.patches import Rectangle
  6. # We just subclass Rectangle so that it can be called with an Axes
  7. # instance, causing the rectangle to update its shape to match the
  8. # bounds of the Axes
  9. class UpdatingRect(Rectangle):
  10.     def __call__(self, ax):
  11.         self.set_bounds(*ax.viewLim.bounds)
  12.         ax.figure.canvas.draw_idle()
  15. # A class that will regenerate a fractal set as we zoom in, so that you
  16. # can actually see the increasing detail.  A box in the left panel will show
  17. # the area to which we are zoomed.
  18. class MandelbrotDisplay(object):
  19.     def __init__(self, h=500, w=500, niter=50, radius=2., power=2):
  20.         self.height = h
  21.         self.width = w
  22.         self.niter = niter
  23.         self.radius = radius
  24.         self.power = power
  26.     def __call__(self, xstart, xend, ystart, yend):
  27.         self.x = np.linspace(xstart, xend, self.width)
  28.         self.y = np.linspace(ystart, yend, self.height).reshape(-1, 1)
  29.         c = self.x + 1.0j * self.y
  30.         threshold_time = np.zeros((self.height, self.width))
  31.         z = np.zeros(threshold_time.shape, dtype=complex)
  32.         mask = np.ones(threshold_time.shape, dtype=bool)
  33.         for i in range(self.niter):
  34.             z[mask] = z[mask]**self.power + c[mask]
  35.             mask = (np.abs(z) < self.radius)
  36.             threshold_time += mask
  37.         return threshold_time
  39.     def ax_update(self, ax):
  40.         ax.set_autoscale_on(False)  # Otherwise, infinite loop
  42.         # Get the number of points from the number of pixels in the window
  43.         dims = ax.patch.get_window_extent().bounds
  44.         self.width = int(dims[2] + 0.5)
  45.         self.height = int(dims[2] + 0.5)
  47.         # Get the range for the new area
  48.         xstart, ystart, xdelta, ydelta = ax.viewLim.bounds
  49.         xend = xstart + xdelta
  50.         yend = ystart + ydelta
  52.         # Update the image object with our new data and extent
  53.         im = ax.images[-1]
  54.         im.set_data(self.__call__(xstart, xend, ystart, yend))
  55.         im.set_extent((xstart, xend, ystart, yend))
  56.         ax.figure.canvas.draw_idle()
  58. md = MandelbrotDisplay()
  59. Z = md(-2., 0.5, -1.25, 1.25)
  61. fig1, (ax1, ax2) = plt.subplots(1, 2)
  62. ax1.imshow(Z, origin='lower', extent=(md.x.min(), md.x.max(), md.y.min(), md.y.max()))
  63. ax2.imshow(Z, origin='lower', extent=(md.x.min(), md.x.max(), md.y.min(), md.y.max()))
  65. rect = UpdatingRect([0, 0], 0, 0, facecolor='None', edgecolor='black', linewidth=1.0)
  66. rect.set_bounds(*ax2.viewLim.bounds)
  67. ax1.add_patch(rect)
  69. # Connect for changing the view limits
  70. ax2.callbacks.connect('xlim_changed', rect)
  71. ax2.callbacks.connect('ylim_changed', rect)
  73. ax2.callbacks.connect('xlim_changed', md.ax_update)
  74. ax2.callbacks.connect('ylim_changed', md.ax_update)
  75. ax2.set_title("Zoom here")
  77. plt.show()

下载这个示例