1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115
| from matplotlib import pyplot as plt from matplotlib.patches import Rectangle, Ellipse from matplotlib.offsetbox import ( AnchoredOffsetbox, AuxTransformBox, DrawingArea, TextArea, VPacker)
class AnchoredText(AnchoredOffsetbox): def __init__(self, s, loc, pad=0.4, borderpad=0.5, prop=None, frameon=True): self.txt = TextArea(s, minimumdescent=False) super().__init__(loc, pad=pad, borderpad=borderpad, child=self.txt, prop=prop, frameon=frameon)
def draw_text(ax): """ Draw a text-box anchored to the upper-left corner of the figure. """ at = AnchoredText("Figure 1a", loc='upper left', frameon=True) at.patch.set_boxstyle("round,pad=0.,rounding_size=0.2") ax.add_artist(at)
class AnchoredDrawingArea(AnchoredOffsetbox): def __init__(self, width, height, xdescent, ydescent, loc, pad=0.4, borderpad=0.5, prop=None, frameon=True): self.da = DrawingArea(width, height, xdescent, ydescent) super().__init__(loc, pad=pad, borderpad=borderpad, child=self.da, prop=None, frameon=frameon)
def draw_circle(ax): """ Draw a circle in axis coordinates """ from matplotlib.patches import Circle ada = AnchoredDrawingArea(20, 20, 0, 0, loc='upper right', pad=0., frameon=False) p = Circle((10, 10), 10) ada.da.add_artist(p) ax.add_artist(ada)
class AnchoredEllipse(AnchoredOffsetbox): def __init__(self, transform, width, height, angle, loc, pad=0.1, borderpad=0.1, prop=None, frameon=True): """ Draw an ellipse the size in data coordinate of the give axes.
pad, borderpad in fraction of the legend font size (or prop) """ self._box = AuxTransformBox(transform) self.ellipse = Ellipse((0, 0), width, height, angle) self._box.add_artist(self.ellipse) super().__init__(loc, pad=pad, borderpad=borderpad, child=self._box, prop=prop, frameon=frameon)
def draw_ellipse(ax): """ Draw an ellipse of width=0.1, height=0.15 in data coordinates """ ae = AnchoredEllipse(ax.transData, width=0.1, height=0.15, angle=0., loc='lower left', pad=0.5, borderpad=0.4, frameon=True)
ax.add_artist(ae)
class AnchoredSizeBar(AnchoredOffsetbox): def __init__(self, transform, size, label, loc, pad=0.1, borderpad=0.1, sep=2, prop=None, frameon=True): """ Draw a horizontal bar with the size in data coordinate of the given axes. A label will be drawn underneath (center-aligned).
pad, borderpad in fraction of the legend font size (or prop) sep in points. """ self.size_bar = AuxTransformBox(transform) self.size_bar.add_artist(Rectangle((0, 0), size, 0, ec="black", lw=1.0))
self.txt_label = TextArea(label, minimumdescent=False)
self._box = VPacker(children=[self.size_bar, self.txt_label], align="center", pad=0, sep=sep)
super().__init__(loc, pad=pad, borderpad=borderpad, child=self._box, prop=prop, frameon=frameon)
def draw_sizebar(ax): """ Draw a horizontal bar with length of 0.1 in data coordinates, with a fixed label underneath. """ asb = AnchoredSizeBar(ax.transData, 0.1, r"1$^{\prime}$", loc='lower center', pad=0.1, borderpad=0.5, sep=5, frameon=False) ax.add_artist(asb)
ax = plt.gca() ax.set_aspect(1.)
draw_text(ax) draw_circle(ax) draw_ellipse(ax) draw_sizebar(ax)
plt.show()
|