Merge pull request #3031 from jrjohansson/fix-axes3d-shade-color

WeatherGod · WeatherGod · commit 67da9edd5e3b · 2014-05-13T10:40:16.000-04:00
avoid np.nan values in colors array returned by axes3d._shade_colors
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1675,11 +1675,13 @@ def _shade_colors(self, color, normals):
         '''
 
         shade = np.array([np.dot(n / proj3d.mod(n), [-1, -1, 0.5])
+                          if proj3d.mod(n) else np.nan
                           for n in normals])
         mask = ~np.isnan(shade)
 
         if len(shade[mask]) > 0:
             norm = Normalize(min(shade[mask]), max(shade[mask]))
+            shade[~mask] = min(shade[mask])
             color = colorConverter.to_rgba_array(color)
             # shape of color should be (M, 4) (where M is number of faces)
             # shape of shade should be (M,)
