Merge pull request matplotlib#2119 from efiring/fix_gci

dmcdougall · dmcdougall · commit e1035ed1d202 · 2013-06-23T15:16:28.000-07:00
pyplot.gci: Look first in current axes, not most recently added axes.
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1204,6 +1204,17 @@ def _gci(self):
         helper for :func:`~matplotlib.pyplot.gci`;
         do not use elsewhere.
         """
+        # Look first for an image in the current Axes:
+        cax = self._axstack.current_key_axes()[1]
+        if cax is None:
+            return None
+        im = cax._gci()
+        if im is not None:
+            return im
+
+        # If there is no image in the current Axes, search for
+        # one in a previously created Axes.  Whether this makes
+        # sense is debatable, but it is the documented behavior.
         for ax in reversed(self.axes):
             im = ax._gci()
             if im is not None:
