Merge pull request #21853 from StefRe/fix/fancyarrow_demo

timhoffm · web-flow · commit 40c77d34f3d8 · 2021-12-03T19:56:45.000+01:00
DOC: Fix Annotation arrow style reference example
diff --git a/examples/text_labels_and_annotations/fancyarrow_demo.py b/examples/text_labels_and_annotations/fancyarrow_demo.py
@@ -7,6 +7,8 @@
 """
 
 import inspect
+import re
+import itertools
 
 import matplotlib.patches as mpatches
 import matplotlib.pyplot as plt
@@ -16,19 +18,19 @@
 nrow = (len(styles) + 1) // ncol
 axs = (plt.figure(figsize=(4 * ncol, 1 + nrow))
        .add_gridspec(1 + nrow, ncol,
-                     wspace=.5, left=.1, right=.9, bottom=0, top=1).subplots())
+                     wspace=.7, left=.1, right=.9, bottom=0, top=1).subplots())
 for ax in axs.flat:
     ax.set_axis_off()
 for ax in axs[0, :]:
     ax.text(0, .5, "arrowstyle",
             transform=ax.transAxes, size="large", color="tab:blue",
             horizontalalignment="center", verticalalignment="center")
-    ax.text(.5, .5, "default parameters",
+    ax.text(.35, .5, "default parameters",
             transform=ax.transAxes,
             horizontalalignment="left", verticalalignment="center")
 for ax, (stylename, stylecls) in zip(axs[1:, :].T.flat, styles.items()):
-    l, = ax.plot(.35, .5, "ok", transform=ax.transAxes)
-    ax.annotate(stylename, (.35, .5), (0, .5),
+    l, = ax.plot(.25, .5, "ok", transform=ax.transAxes)
+    ax.annotate(stylename, (.25, .5), (-0.1, .5),
                 xycoords="axes fraction", textcoords="axes fraction",
                 size="large", color="tab:blue",
                 horizontalalignment="center", verticalalignment="center",
@@ -37,8 +39,12 @@
                     color="k", shrinkA=5, shrinkB=5, patchB=l,
                 ),
                 bbox=dict(boxstyle="square", fc="w"))
-    ax.text(.5, .5,
-            str(inspect.signature(stylecls))[1:-1].replace(", ", "\n"),
+    # wrap at every nth comma (n = 1 or 2, depending on text length)
+    s = str(inspect.signature(stylecls))[1:-1]
+    n = 2 if s.count(',') > 3 else 1
+    ax.text(.35, .5,
+            re.sub(', ', lambda m, c=itertools.count(1): m.group()
+                   if next(c) % n else '\n', s),
             transform=ax.transAxes,
             horizontalalignment="left", verticalalignment="center")
 
