matplotlib · jlecoeur · Mar 14, 2017 · Mar 14, 2017 · Mar 14, 2017 · Mar 14, 2017
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -263,7 +263,8 @@ def _get_legend_handles(self, legend_handler_map=None):
 
         """
         handles_original = (self.lines + self.patches +
-                            self.collections + self.containers)
+                            self.collections + self.containers +
+                                self.texts)
         handler_map = mlegend.Legend.get_default_handler_map()
 
         if legend_handler_map is not None:

diff --git a/lib/matplotlib/legend.py b/lib/matplotlib/legend.py
@@ -36,7 +36,8 @@
 from matplotlib.cbook import is_string_like, silent_list, is_hashable
 from matplotlib.font_manager import FontProperties
 from matplotlib.lines import Line2D
-from matplotlib.patches import Patch, Rectangle, Shadow, FancyBboxPatch
+from matplotlib.patches import Patch, Rectangle, Shadow, FancyBboxPatch, FancyArrowPatch
+from matplotlib.text import Text, Annotation
 from matplotlib.collections import (LineCollection, RegularPolyCollection,
                                     CircleCollection, PathCollection,
                                     PolyCollection)
@@ -493,7 +494,10 @@ def _approx_text_height(self, renderer=None):
             update_func=legend_handler.update_from_first_child),
         tuple: legend_handler.HandlerTuple(),
         PathCollection: legend_handler.HandlerPathCollection(),
-        PolyCollection: legend_handler.HandlerPolyCollection()
+        PolyCollection: legend_handler.HandlerPolyCollection(),
+        Text:legend_handler.HandlerText(),
+        FancyArrowPatch:legend_handler.HandlerFancyArrowPatch(),
+        Annotation:legend_handler.HandlerAnnotation(),
         }
 
     # (get|set|update)_default_handler_maps are public interfaces to

diff --git a/lib/matplotlib/legend_handler.py b/lib/matplotlib/legend_handler.py
@@ -34,11 +34,13 @@ def legend_artist(self, legend, orig_handle, fontsize, handlebox):
 import numpy as np
 
 from matplotlib.lines import Line2D
-from matplotlib.patches import Rectangle
+from matplotlib.patches import Rectangle, FancyArrowPatch
+from matplotlib.text import Text, Annotation
 import matplotlib.collections as mcoll
 import matplotlib.colors as mcolors
 
 
+
 def update_from_first_child(tgt, src):
     tgt.update_from(src.get_children()[0])
 
@@ -256,6 +258,21 @@ def create_artists(self, legend, orig_handle,
         return [p]
 
 
+class HandlerFancyArrowPatch(HandlerPatch):
+    """
+    Handler for FancyArrowPatch instances.
+    """
+    def _create_patch(self, legend, orig_handle,
+                      xdescent, ydescent, width, height, fontsize):
+        arrow = FancyArrowPatch( [-xdescent,
+                                 -ydescent + height / 2],
+                                [-xdescent + width,
+                                 -ydescent + height / 2],
+                                mutation_scale=width / 3)
+        arrow.set_arrowstyle(orig_handle.get_arrowstyle())
+        return arrow
+
+
 class HandlerLineCollection(HandlerLine2D):
     """
     Handler for LineCollection instances.
@@ -578,18 +595,25 @@ class HandlerTuple(HandlerBase):
         The number of sections to divide the legend area into.  If None,
         use the length of the input tuple. Default is 1.
 
-
     pad : float, optional
         If None, fall back to `legend.borderpad` as the default.
         In units of fraction of font size. Default is None.
 
-
+    width_ratios : tuple, optional
+        The relative width of sections. Must be of length ndivide.
+        If None, all sections will have the same width. Default is None.
 
     """
-    def __init__(self, ndivide=1, pad=None, **kwargs):
+    def __init__(self, ndivide=1, pad=None, width_ratios=None, **kwargs):
 
         self._ndivide = ndivide
         self._pad = pad
+
+        if (width_ratios is not None) and (len(width_ratios) == ndivide):
+            self._width_ratios = width_ratios
+        else:
+            self._width_ratios = None
+
         HandlerBase.__init__(self, **kwargs)
 
     def create_artists(self, legend, orig_handle,
@@ -608,10 +632,16 @@ def create_artists(self, legend, orig_handle,
         else:
             pad = self._pad * fontsize
 
-        if ndivide > 1:
-            width = (width - pad*(ndivide - 1)) / ndivide
+        if self._width_ratios is not None:
+            sumratios = sum(self._width_ratios)
+            widths = [(width - pad * (ndivide - 1)) * ratio / sumratios
+                      for ratio in self._width_ratios]
+        else:
+            widths = [(width - pad * (ndivide - 1)) / ndivide
+                      for _ in range(ndivide)]
+        widths_cycle = cycle(widths)
 
-        xds = [xdescent - (width + pad) * i for i in range(ndivide)]
+        xds = [xdescent - (widths[-i-1] + pad) * i for i in range(ndivide)]
         xds_cycle = cycle(xds)
 
         a_list = []
@@ -620,7 +650,8 @@ def create_artists(self, legend, orig_handle,
             _a_list = handler.create_artists(legend, handle1,
                                              six.next(xds_cycle),
                                              ydescent,
-                                             width, height,
+                                             six.next(widths_cycle),
+                                             height,
                                              fontsize,
                                              trans)
             a_list.extend(_a_list)
@@ -661,3 +692,59 @@ def create_artists(self, legend, orig_handle,
         self.update_prop(p, orig_handle, legend)
         p.set_transform(trans)
         return [p]
+
+
+class HandlerText(HandlerBase):
+    """
+    Handler for Text instances.
+    """
+    def create_artists(self, legend, orig_handle,
+                       xdescent, ydescent, width, height, fontsize, trans):
        t = Text(x=-xdescent + width / 3,
+                 y=-ydescent + height / 4,
+                 text="a")
+
+        # Use original text if it is short
+        text = orig_handle.get_text()
+        if len(text) < 2:
+            t.set_text(text)
+
+        # Copy text attributes, except fontsize
+        self.update_prop(t, orig_handle, legend)
+        t.set_transform(trans)
+        t.set_fontsize(2 * fontsize / 3)
+
+        return [t]
+
+
+class HandlerAnnotation(HandlerBase):
+    """
+    Handler for FancyArrowPatch instances.
+        """
+    def create_artists(self, legend, orig_handle,
+                       xdescent, ydescent, width, height, fontsize, trans):
+        if (orig_handle.arrow_patch is not None) and (orig_handle.get_text() is not ""):
+            # Draw a tuple (text, arrow)
+            handler = HandlerTuple(ndivide=2, pad=None, width_ratios=[1,4])
+
+            # Create a Text instance from annotation text
+            text_handle = Text(text=orig_handle.get_text())
+            text_handle.update_from(orig_handle)
+            handle  = (text_handle, orig_handle.arrow_patch)
+        elif orig_handle.arrow_patch is not None:
+            # Arrow without text
+            handler = HandlerFancyArrowPatch()
+            handle  = orig_handle.arrow_patch
+        elif orig_handle.get_text() is not "":
+            # Text without arrow
+            handler = HandlerText()
+            handle  = orig_handle
+        else:
+            # No text, no arrow
+            handler = HandlerPatch()
+            handle  = Rectangle(xy=[0,0],width=0,height=0,color='w')
+
+        return handler.create_artists(legend, handle,
+                                       xdescent, ydescent,
+                                       width, height,
+                                       fontsize, trans)