FIX: set sca when re-using as Axes with pyplot.subplot

tacaswell · tacaswell · commit 385e1a5fcca6 · 2021-02-02T20:12:05.000-05:00
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -1232,6 +1232,7 @@ def subplot(*args, **kwargs):
     # If no existing axes matches, then create a new one.
     if ax is None or getattr(ax, '_init_kwargs', {}) != kwargs:
         ax = fig.add_subplot(*args, **kwargs)
+    fig.sca(ax)
 
     bbox = ax.bbox
     axes_to_delete = []
diff --git a/lib/matplotlib/tests/test_pyplot.py b/lib/matplotlib/tests/test_pyplot.py
@@ -153,3 +153,12 @@ def test_nested_ion_ioff():
     with plt.ioff():
         plt.ion()
     assert not mpl.is_interactive()
+
+
+def test_subplot_reuse():
+    ax1 = plt.subplot(121)
+    assert ax1 is plt.gca()
+    ax2 = plt.subplot(122)
+    assert ax2 is plt.gca()
+    ax1 = plt.subplot(121)
+    assert ax1 is plt.gca()
