Merge pull request #6622 from data-exp-lab/strip-units-imshow

tacaswell · web-flow · commit f8ea2c51224e · 2016-06-22T19:49:25.000-04:00
FIX: ensure masked array data is an ndarray
diff --git a/lib/matplotlib/colors.py b/lib/matplotlib/colors.py
@@ -925,7 +925,9 @@ def __call__(self, value, clip=None):
                 result = np.ma.array(np.clip(result.filled(vmax), vmin, vmax),
                                      mask=mask)
             # ma division is very slow; we can take a shortcut
-            resdat = result.data
+            # use np.asarray so data passed in as an ndarray subclass are
+            # interpreted as an ndarray. See issue #6622.
+            resdat = np.asarray(result.data)
             resdat -= vmin
             resdat /= (vmax - vmin)
             result = np.ma.array(resdat, mask=result.mask, copy=False)
