matplotlib · xiaoshiqi · Feb 16, 2018 · Feb 16, 2018 · Feb 16, 2018 · Feb 17, 2018
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -15,6 +15,7 @@
 import six
 from six.moves import map, zip, reduce

+import types
 from collections import defaultdict
 import math
 import warnings
@@ -2364,6 +2365,60 @@ def scatter(self, xs, ys, zs=0, zdir='z', s=20, c=None, depthshade=True,
        art3d.patch_collection_2d_to_3d(patches, zs=zs, zdir=zdir,
                                        depthshade=depthshade)

+        def set_facecolor3d(self, c):
+            """
+            Set the facecolor(s) of the collection.  *c* can be a
+            matplotlib color spec (all patches have same color), or a
+            sequence of specs; if it is a sequence the patches will
+            cycle through the sequence.
+
+            If *c* is 'none', the patch will not be filled.
+
+            ACCEPTS: matplotlib color spec or sequence of specs
+            """
+            self.set_facecolor2d(c)
+
+            c = self.get_facecolor().copy()    # new color
+            length_old = self._facecolor3d.shape[0]
+            length_new = c.shape[0]
+            # reset alpha value to original
+            if (length_new > length_old):
+                c[:, 3] = self._facecolor3d[0, 3]
+            elif (length_new == length_old):
+                c[:, 3] = self._facecolor3d[:, 3]
+            self._facecolor3d = c
+
+        def set_edgecolor3d(self, c):
+            """
+            Set the edgecolor(s) of the collection. *c* can be a
+            matplotlib color spec (all patches have same color), or a
+            sequence of specs; if it is a sequence the patches will
+            cycle through the sequence.
+
+            If *c* is 'face', the edge color will always be the same as
+            the face color.  If it is 'none', the patch boundary will not
+            be drawn.
+
+            ACCEPTS: matplotlib color spec or sequence of specs
+            """
+            self.set_edgecolor2d(c)
+
+            c = self.get_edgecolor().copy()    # new color
+            length_old = self._edgecolor3d.shape[0]
+            length_new = c.shape[0]
+
+            # reset alpha value to original
+            if (length_new > length_old):
+                c[:, 3] = self._edgecolor3d[0, 3]
+            elif (length_new == length_old):
+                c[:, 3] = self._edgecolor3d[:, 3]
+            self._edgecolor3d = c
+
+        patches.set_facecolor2d = patches.set_facecolor
+        patches.set_facecolor = types.MethodType(set_facecolor3d, patches)
+        patches.set_edgecolor2d = patches.set_edgecolor
+        patches.set_edgecolor = types.MethodType(set_edgecolor3d, patches)
+
        if self._zmargin < 0.05 and xs.size > 0:
            self.set_zmargin(0.05)


diff --git a/lib/mpl_toolkits/tests/baseline_images/test_mplot3d/scatter3d_change_color.png b/lib/mpl_toolkits/tests/baseline_images/test_mplot3d/scatter3d_change_color.png
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -175,6 +175,26 @@ def test_scatter3d_color():
               color='b', marker='s')


+@image_comparison(baseline_images=['scatter3d_change_color'], remove_text=True,
+                  style='mpl20', extensions=['png'])
+def test_scatter3d_change_color():
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    ax.xaxis.set_major_formatter(plt.NullFormatter())
+    ax.yaxis.set_major_formatter(plt.NullFormatter())
+    ax.zaxis.set_major_formatter(plt.NullFormatter())
+    sca1 = ax.scatter(0, 0, 0, color='r', marker='o')
+    sca2 = ax.scatter(np.arange(1, 3), np.arange(1, 3), np.arange(1, 3),
+                      color=['orange', 'green'], marker='^')
+    sca3 = ax.scatter(np.arange(3, 6), np.arange(3, 6), np.arange(3, 6),
+                      color='#FF0000', marker='s')
+
+    sca1.set_color("green")
+    sca2.set_facecolor([0, 1, 1])
+    sca2.set_edgecolor('face')
+    sca3.set_facecolor(["white", "#ABCDEF", [0, 0, 0]])
+
+
 @image_comparison(baseline_images=['surface3d'], remove_text=True)
 def test_surface3d():
    fig = plt.figure()