fix error in scatter

patniharshit · patniharshit · commit 13627d336881 · 2017-07-23T16:17:37.000+05:30
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -3979,17 +3979,6 @@ def scatter(self, x, y, s=None, c=None, marker=None, cmap=None, norm=None,
         else:
             try:
                 c_array = np.asanyarray(c, dtype=float)
-                if c_array.ndim > 1:
-                    if cmap is None:
-                        cmap = mcolors.BivariateColormap()
-                    if norm is None:
-                        norm = mcolors.BivariateNorm()
-                    c_array = norm(c_array)
-                    c_array[0] = c_array[0] * (cmap.N-1)
-                    c_array[1] = c_array[1] * (cmap.N-1)
-                    c_array = c_array.astype(int)
-                    c_array = c_array[0] + cmap.N * c_array[1]
-                    norm = mcolors.NoNorm()
                 if c_array.shape in xy_shape:
                     c = np.ma.ravel(c_array)
                 else:
@@ -4054,10 +4043,8 @@ def scatter(self, x, y, s=None, c=None, marker=None, cmap=None, norm=None,
         collection.update(kwargs)
 
         if colors is None:
-            isNorm = isinstance(norm, (mcolors.Normalize, mcolors.BivariateNorm))
-            if norm is not None and not isNorm:
-                msg = "'norm' must be an instance of 'mcolors.Normalize' " \
-                      "or 'mcolors.BivariateNorm'"
+            if norm is not None and not isinstance(norm, mcolors.Normalize):
+                msg = "'norm' must be an instance of 'mcolors.Normalize'"
                 raise ValueError(msg)
             collection.set_array(np.asarray(c))
             collection.set_cmap(cmap)
