Avoid warning on nan input to log-scaled plots.

anntzer · anntzer · commit 63af7a775779 · 2016-09-20T00:29:57.000-07:00
diff --git a/lib/matplotlib/scale.py b/lib/matplotlib/scale.py
@@ -98,7 +98,8 @@ def __init__(self, nonpos):
             self._fill_value = 1e-300
 
     def transform_non_affine(self, a):
-        a = np.where(a <= 0, self._fill_value, a)
+        with np.errstate(invalid="ignore"):
+            a = np.where(a <= 0, self._fill_value, a)
         np.log(a, a)
         a /= np.log(self.base)
         a += 1
@@ -435,8 +436,9 @@ def __init__(self, nonpos):
 
     def transform_non_affine(self, a):
         """logit transform (base 10), masked or clipped"""
-        a = np.select(
-            [a <= 0, a >= 1], [self._fill_value, 1 - self._fill_value], a)
+        with np.errstate(invalid="ignore"):
+            a = np.select(
+                [a <= 0, a >= 1], [self._fill_value, 1 - self._fill_value], a)
         return np.log10(1.0 * a / (1.0 - a))
 
     def inverted(self):
