Return the list of added Poly3DCollections

eric-wieser · eric-wieser · commit be532e3be5d2 · 2016-06-20T09:20:57.000+01:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -2666,6 +2666,11 @@ def voxels(self, filled, color=None):
             Either a single value or an array the same shape as filled,
             indicating what color to draw the faces of the voxels. If None,
             plot all voxels in the same color, the next in the color sequence
+
+        Returns
+        -------
+        list of `Poly3DCollection`
+            The square faces which were plotted, in an arbitrary order
         """
         # check dimensions, and deal with a single color
         if filled.ndim != 3:
@@ -2689,6 +2694,7 @@ def voxels(self, filled, color=None):
             [0, filled.shape[2]]
         )
 
+        polygons = []
 
         # points lying on corners of a square
         square = np.array([
@@ -2703,6 +2709,7 @@ def boundary_found(corners, color):
             poly = art3d.Poly3DCollection([corners])
             poly.set_facecolor(color)
             self.add_collection3d(poly)
+            polygons.append(poly)
 
         def permutation_matrices(n):
             """ Generator of cyclic permutation matices """
@@ -2746,6 +2753,8 @@ def permutation_matrices(n):
                     if filled[ik]:
                         boundary_found(pk2 + square.dot(permute.T), color[ik])
 
+        return polygons
+
 
 def get_test_data(delta=0.05):
     '''
