FIX: make sure scalarmappable updates are handleded correctly in 3D

tacaswell · tacaswell · commit 10361938593e · 2020-11-10T16:55:26.000-05:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -592,6 +592,7 @@ def set_linewidth(self, lw):
 
     @cbook._delete_parameter('3.4', 'renderer')
     def do_3d_projection(self, renderer=None):
+        self.update_scalarmappable()
         xs, ys, zs = self._offsets3d
         vxs, vys, vzs, vis = proj3d.proj_transform_clip(xs, ys, zs,
                                                         self.axes.M)
@@ -634,6 +635,16 @@ def do_3d_projection(self, renderer=None):
 
         return np.min(vzs) if vzs.size else np.nan
 
+    def update_scalarmappable(self):
+        copy_state = self._update_dict['array']
+        ret = super().update_scalarmappable()
+        if copy_state:
+            if self._is_filled:
+                self._facecolors3d = self._facecolors
+            elif self._is_stroked:
+                self._edgecolors3d = self._edgecolors
+        return ret
+
 
 def patch_collection_2d_to_3d(col, zs=0, zdir='z', depthshade=True):
     """
