add public set_loc method for

cocolato · cocolato · commit 93d288e5b8d0 · 2023-06-07T19:37:37.000Z
diff --git a/lib/matplotlib/legend.py b/lib/matplotlib/legend.py
@@ -686,6 +686,19 @@ def _set_artist_props(self, a):
 
         a.set_transform(self.get_transform())
 
+    def set_loc(self, loc):
+        """
+        Set the location of the legend.
+
+        Parameters
+        ----------
+        alignment : {'upper left', 'upper right', 'lower left', 'lower right',
+                    'upper center', 'lower center', 'center left', 'center right'
+                    'center', 'right', 'best'}.
+        """
+        _loc_code = self.codes.get(loc, 'best')
+        self._set_loc(_loc_code)
+
     def _set_loc(self, loc):
         # find_offset function will be provided to _legend_box and
         # _legend_box will draw itself at the location of the return
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -755,6 +755,15 @@ def test_legend_alignment(alignment):
     assert leg.get_alignment() == alignment
 
 
+@pytest.mark.parametrize('loc', ('center', 'best'))
+def test_legend_set_loc(loc):
+    fig, ax = plt.subplots()
+    ax.plot(range(10), label='test')
+    leg = ax.legend()
+    leg.set_loc(loc)
+    assert leg._get_loc() == mlegend.Legend.codes[loc]
+
+
 @pytest.mark.parametrize('alignment', ('center', 'left', 'right'))
 def test_legend_set_alignment(alignment):
     fig, ax = plt.subplots()
