matplotlib
diff --git a/‎doc/users/next_whats_new/boxplot_legend_support.rst
Copy file name to clipboardExpand all lines: doc/users/next_whats_new/boxplot_legend_support.rst
+26-5Lines changed: 26 additions & 5 deletions b/‎doc/users/next_whats_new/boxplot_legend_support.rst
Copy file name to clipboardExpand all lines: doc/users/next_whats_new/boxplot_legend_support.rst
+26-5Lines changed: 26 additions & 5 deletions
diff --git a/‎lib/matplotlib/axes/_axes.py
Copy file name to clipboardExpand all lines: lib/matplotlib/axes/_axes.py
+10-16Lines changed: 10 additions & 16 deletions b/‎lib/matplotlib/axes/_axes.py
Copy file name to clipboardExpand all lines: lib/matplotlib/axes/_axes.py
+10-16Lines changed: 10 additions & 16 deletions
diff --git a/‎lib/matplotlib/tests/test_legend.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_legend.py
+7-10Lines changed: 7 additions & 10 deletions b/‎lib/matplotlib/tests/test_legend.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_legend.py
+7-10Lines changed: 7 additions & 10 deletions
@@ -1,14 +1,14 @@
 Legend support for Boxplot
 ~~~~~~~~~~~~~~~~~~~~~~~~~~
 Boxplots now support a *label* parameter to create legend entries.
-Legend labels can be passed as a string to each individual boxplot
-or as a list of strings to label multiple boxes in a single
-boxplot call.
+
+Legend labels can be passed as a list of strings to label multiple boxes in a single
+boxplot call:
 
 
 .. plot::
     :include-source: true
-    :alt: Example of creating 3 boxplots and assigning legend labels with keywords.
+    :alt: Example of creating 3 boxplots and assigning legend labels as a sequence.
 
     import matplotlib.pyplot as plt
     import numpy as np
@@ -34,4 +34,25 @@ boxplot call.
         patch.set_facecolor(color)
 
     ax.legend()
-    plt.show()
+
+
+Or as a single string to each individual boxplot. Note that the plot without a
+patch artist will get its legend handle from the median line:
+
+.. plot::
+    :include-source: true
+    :alt: Example of creating 2 boxplots and assigning each legend label as a string.
+
+    fig, ax = plt.subplots()
+    data_A = np.random.random((100, 3))
+    data_B = np.random.random((100, 3)) + 0.2
+    pos = np.arange(3)
+
+    bplot2 = ax.boxplot(data_A, positions=pos - 0.2, label='Box A')
+    bplot3 = ax.boxplot(data_B, positions=pos + 0.2, patch_artist=True, label='Box B')
+
+    # fill with colors
+    for patch in bplot3['boxes']:
+        patch.set_facecolor('lightblue')
+
+    ax.legend()
@@ -4408,22 +4408,16 @@ def do_patch(xs, ys, **kwargs):
 
         # Set legend labels
         if label:
-            box_or_med = boxes if showbox else medians
-            for index, element in enumerate(box_or_med):
-                if cbook.is_scalar_or_string(label):
-                    # If there are multiple boxes, the label
-                    # should only be applied to the first box.
-                    if len(box_or_med) > len(label):
-                        element.set_label(label)
-                        break
-                    else:
-                        element.set_label(label)
-                else:  # label is a sequence
-                    if len(box_or_med) == len(label):
-                        element.set_label(label[index])
-                    else:
-                        raise ValueError("There must be an equal number of legend"
-                                         " labels and boxplots.")
+            box_or_med = boxes if showbox and patch_artist else medians
+            if cbook.is_scalar_or_string(label):
+                # assign the label only to the first box
+                box_or_med[0].set_label(label)
+            else:  # label is a sequence
+                if len(box_or_med) != len(label):
+                    raise ValueError("There must be an equal number of legend"
+                                     " labels and boxplots.")
+                for artist, lbl in zip(box_or_med, label):
+                    artist.set_label(lbl)
 
         if manage_ticks:
             axis_name = "x" if vert else "y"
 
@@ -1457,21 +1457,19 @@ def test_boxplot_legend_labels():
     np.random.seed(19680801)
     data = np.random.random((10, 4))
     fig, axs = plt.subplots(nrows=1, ncols=4)
-    legend_labels = ['A', 'B', 'C', 'D']
+    legend_labels = ['box A', 'box B', 'box C', 'box D']
 
     # Testing legend labels and patch passed to legend.
     bp1 = axs[0].boxplot(data, patch_artist=True, label=legend_labels)
-    for i, v in enumerate(bp1['boxes']):
-        assert v.get_label() == legend_labels[i]
+    assert [v.get_label() for v in bp1['boxes']] == legend_labels
     handles, labels = axs[0].get_legend_handles_labels()
     assert labels == legend_labels
     assert all(isinstance(h, mpl.patches.PathPatch) for h in handles)
 
     # Testing legend without `box`.
     bp2 = axs[1].boxplot(data, label=legend_labels, showbox=False)
     # Without a box, The legend entries should be passed from the medians.
-    for i, v in enumerate(bp2['medians']):
-        assert v.get_label() == legend_labels[i]
+    assert [v.get_label() for v in bp2['medians']] == legend_labels
     handles, labels = axs[1].get_legend_handles_labels()
     assert labels == legend_labels
     assert all(isinstance(h, mpl.lines.Line2D) for h in handles)
@@ -1480,8 +1478,7 @@ def test_boxplot_legend_labels():
     with pytest.raises(ValueError, match='There must be an equal number'):
         bp3 = axs[2].boxplot(data, label=legend_labels[:-1])
 
-    # Of 3 identical boxes, test that only the first box gets a label.
-    new_data = np.random.random((100, 3))
-    bp4 = axs[3].boxplot(new_data, label='A')
-    assert bp4['boxes'][0].get_label() == 'A' and all(x.get_label() != 'A'
-                                                      for x in bp4['boxes'][1:])
+    # Test that for a string label, only the first box gets a label.
+    bp4 = axs[3].boxplot(data, label='box A')
+    assert bp4['medians'][0].get_label() == 'box A'
+    assert all(x.get_label().startswith("_") for x in bp4['medians'][1:])