FIX: Apply aspect before drawing starts

jklymak · jklymak · commit 69fd1fc5c070 · 2018-09-21T20:53:39.000-07:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -2553,12 +2553,6 @@ def draw(self, renderer=None, inframe=False):
 
         # prevent triggering call backs during the draw process
         self._stale = True
-        locator = self.get_axes_locator()
-        if locator:
-            pos = locator(self, renderer)
-            self.apply_aspect(pos)
-        else:
-            self.apply_aspect()
 
         artists = self.get_children()
         artists.remove(self.patch)
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1627,6 +1627,23 @@ def draw(self, renderer):
              if not artist.get_animated()),
             key=lambda artist: artist.get_zorder())
 
+        for ax in self.axes:
+            locator = ax.get_axes_locator()
+            if locator:
+                pos = locator(ax, renderer)
+                ax.apply_aspect(pos)
+            else:
+                ax.apply_aspect()
+
+            for child in ax.get_children():
+                if hasattr(child, 'apply_aspect'):
+                    locator = child.get_axes_locator()
+                    if locator:
+                        pos = locator(child, renderer)
+                        child.apply_aspect(pos)
+                    else:
+                        child.apply_aspect()
+
         try:
             renderer.open_group('figure')
             if self.get_constrained_layout() and self.axes:
