Return a scalar when passed in a scalar

jkseppan · jkseppan · commit b069a942ab15 · 2015-09-03T15:33:13.000+03:00
Since possibly some earlier versions of numpy returned a scalar,
wrap the value in atleast_1d before indexing.
diff --git a/lib/matplotlib/colors.py b/lib/matplotlib/colors.py
@@ -1026,6 +1026,8 @@ def __call__(self, value, clip=None):
             result = ma.masked_array(np.interp(result, x, y),
                                      mask=ma.getmask(result))
 
+        if is_scalar:
+            result = np.atleast_1d(result)[0]
         return result
 
     def autoscale_None(self, A):
