Optimized plotting of NaNs in QuadMesh

holoviz · Jan 8, 2018 · 9a65e73 · 9a65e73
1 parent 8b62fe9
commit 9a65e73
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/holoviews/plotting/bokeh/raster.py b/holoviews/plotting/bokeh/raster.py
@@ -194,7 +194,12 @@ def get_data(self, element, ranges, style):
                     for d in dims]
             X, Y = colormesh(X, Y)
             zvals = zdata.T.flatten() if self.invert_axes else zdata.flatten()
-            data = {'xs': list(X), 'ys': list(Y), z.name: zvals}
+            XS, YS = [], []
+            for x, y, zval in zip(X, Y, zvals):
+                if np.isfinite(zval):
+                    XS.append(x)
+                    YS.append(y)
+            data = {'xs': XS, 'ys': YS, z.name: zvals[np.isfinite(zvals)]}
         else:
             xc, yc = (element.interface.coords(element, x, edges=True),
                       element.interface.coords(element, y, edges=True))