图中插入注释框

图中插入注释框示例

  1. import matplotlib.pyplot as plt
  2. import numpy as np
  4. from matplotlib.patches import Circle
  5. from matplotlib.offsetbox import (TextArea, DrawingArea, OffsetImage,
  6.                                   AnnotationBbox)
  7. from matplotlib.cbook import get_sample_data
  10. if 1:
  11.     fig, ax = plt.subplots()
  13.     # Define a 1st position to annotate (display it with a marker)
  14.     xy = (0.5, 0.7)
  15.     ax.plot(xy[0], xy[1], ".r")
  17.     # Annotate the 1st position with a text box ('Test 1')
  18.     offsetbox = TextArea("Test 1", minimumdescent=False)
  20.     ab = AnnotationBbox(offsetbox, xy,
  21.                         xybox=(-20, 40),
  22.                         xycoords='data',
  23.                         boxcoords="offset points",
  24.                         arrowprops=dict(arrowstyle="->"))
  25.     ax.add_artist(ab)
  27.     # Annotate the 1st position with another text box ('Test')
  28.     offsetbox = TextArea("Test", minimumdescent=False)
  30.     ab = AnnotationBbox(offsetbox, xy,
  31.                         xybox=(1.02, xy[1]),
  32.                         xycoords='data',
  33.                         boxcoords=("axes fraction", "data"),
  34.                         box_alignment=(0., 0.5),
  35.                         arrowprops=dict(arrowstyle="->"))
  36.     ax.add_artist(ab)
  38.     # Define a 2nd position to annotate (don't display with a marker this time)
  39.     xy = [0.3, 0.55]
  41.     # Annotate the 2nd position with a circle patch
  42.     da = DrawingArea(20, 20, 0, 0)
  43.     p = Circle((10, 10), 10)
  44.     da.add_artist(p)
  46.     ab = AnnotationBbox(da, xy,
  47.                         xybox=(1.02, xy[1]),
  48.                         xycoords='data',
  49.                         boxcoords=("axes fraction", "data"),
  50.                         box_alignment=(0., 0.5),
  51.                         arrowprops=dict(arrowstyle="->"))
  53.     ax.add_artist(ab)
  55.     # Annotate the 2nd position with an image (a generated array of pixels)
  56.     arr = np.arange(100).reshape((10, 10))
  57.     im = OffsetImage(arr, zoom=2)
  58.     im.image.axes = ax
  60.     ab = AnnotationBbox(im, xy,
  61.                         xybox=(-50., 50.),
  62.                         xycoords='data',
  63.                         boxcoords="offset points",
  64.                         pad=0.3,
  65.                         arrowprops=dict(arrowstyle="->"))
  67.     ax.add_artist(ab)
  69.     # Annotate the 2nd position with another image (a Grace Hopper portrait)
  70.     fn = get_sample_data("grace_hopper.png", asfileobj=False)
  71.     arr_img = plt.imread(fn, format='png')
  73.     imagebox = OffsetImage(arr_img, zoom=0.2)
  74.     imagebox.image.axes = ax
  76.     ab = AnnotationBbox(imagebox, xy,
  77.                         xybox=(120., -80.),
  78.                         xycoords='data',
  79.                         boxcoords="offset points",
  80.                         pad=0.5,
  81.                         arrowprops=dict(
  82.                             arrowstyle="->",
  83.                             connectionstyle="angle,angleA=0,angleB=90,rad=3")
  84.                         )
  86.     ax.add_artist(ab)
  88.     # Fix the display limits to see everything
  89.     ax.set_xlim(0, 1)
  90.     ax.set_ylim(0, 1)
  92.     plt.show()

下载这个示例