Fix problem with legend if data has NaN's

demotu · demotu · commit 304ddd302571 · 2014-10-03T08:06:01.000-03:00
If data has NaN's and plot has a legend, matplotlib throws the following
RuntimeWarning:
...\matplotlib\transforms.py:651: RuntimeWarning: invalid value
encountered in sign
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -647,6 +647,7 @@ def count_contains(self, vertices):
         if len(vertices) == 0:
             return 0
         vertices = np.asarray(vertices)
+        vertices = vertices[~np.isnan(np.sum(vertices, 1)), :]  # handle NaN's
         x0, y0, x1, y1 = self._get_extents()
         dx0 = np.sign(vertices[:, 0] - x0)
         dy0 = np.sign(vertices[:, 1] - y0)
