set_aspect('equal') now adopts current box aspect

Author: scottshambaugh

lib/mpl_toolkits/mplot3d/axes3d.py

@@ -324,16 +324,17 @@ def set_aspect(self, aspect, adjustable=None, anchor=None, share=False):
             aspect, adjustable=adjustable, anchor=anchor, share=share)
 
         if aspect == 'equal':
-            self.set_box_aspect((1., 1., 1.))
             v_intervals = np.vstack((self.xaxis.get_view_interval(),
                                      self.yaxis.get_view_interval(),
                                      self.zaxis.get_view_interval()))
             mean = np.mean(v_intervals, axis=1)
             delta = np.max(np.ptp(v_intervals, axis=1))
+            deltas = delta * self._box_aspect / min(self._box_aspect)
+
+            self.set_xlim3d(mean[0] - deltas[0] / 2., mean[0] + deltas[0] / 2.)
+            self.set_ylim3d(mean[1] - deltas[1] / 2., mean[1] + deltas[1] / 2.)
+            self.set_zlim3d(mean[2] - deltas[2] / 2., mean[2] + deltas[2] / 2.)
 
-            self.set_xlim3d(mean[0] - delta / 2., mean[0] + delta / 2.)
-            self.set_ylim3d(mean[1] - delta / 2., mean[1] + delta / 2.)
-            self.set_zlim3d(mean[2] - delta / 2., mean[2] + delta / 2.)
 
     def set_box_aspect(self, aspect, *, zoom=1):
         """