get_zaxis method for 3d axes

ImportanceOfBeingErnest · ImportanceOfBeingErnest · commit 0a70730fa721 · 2018-06-06T16:18:07.000+02:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -192,19 +192,24 @@ def set_top_view(self):
 
     def _init_axis(self):
         '''Init 3D axes; overrides creation of regular X/Y axes'''
-        self.w_xaxis = axis3d.XAxis('x', self.xy_viewLim.intervalx,
+        self.xaxis = axis3d.XAxis('x', self.xy_viewLim.intervalx,
                                     self.xy_dataLim.intervalx, self)
-        self.xaxis = self.w_xaxis
-        self.w_yaxis = axis3d.YAxis('y', self.xy_viewLim.intervaly,
+        self.yaxis = axis3d.YAxis('y', self.xy_viewLim.intervaly,
                                     self.xy_dataLim.intervaly, self)
-        self.yaxis = self.w_yaxis
-        self.w_zaxis = axis3d.ZAxis('z', self.zz_viewLim.intervalx,
+        self.zaxis = axis3d.ZAxis('z', self.zz_viewLim.intervalx,
                                     self.zz_dataLim.intervalx, self)
-        self.zaxis = self.w_zaxis
+        # Provide old aliases
+        self.w_xaxis = self.xaxis
+        self.w_yaxis = self.yaxis
+        self.w_zaxis = self.zaxis
 
         for ax in self.xaxis, self.yaxis, self.zaxis:
             ax.init3d()
 
+    def get_zaxis(self):
+        '''Return the :class:`ZAxis<~.axis3d.Axis>` instance.'''
+        return self.zaxis
+
     def get_children(self):
         return [self.zaxis] + super().get_children()
 
@@ -2435,18 +2440,18 @@ def bar3d(self, x, y, z, dx, dy, dz, color=None,
 
         zsort : str, optional
             The z-axis sorting scheme passed onto
-            :func:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection`
+            :class:`~.art3d.Poly3DCollection`
 
         shade : bool, optional (default = True)
             When true, this shades the dark sides of the bars (relative
             to the plot's source of light).
 
-        Any additional keyword arguments are passed onto
-        :func:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection`
+        **kwargs : Any additional keyword arguments are passed onto
+            :class:`~.art3d.Poly3DCollection`.
 
         Returns
         -------
-        collection : Poly3DCollection
+        collection : `~.art3d.Poly3DCollection`
             A collection of three dimensional polygons representing
             the bars.
         """
