Fix kde2d for datasets with nonfinite values

Using the method kde2d will no longer cause an error if x/y contain non-finite values.
LKremer · Feb 13, 2020 · 0264094 · 0264094
1 parent 7c6cb1d
commit 0264094
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/R/geom_pointdensity.R b/R/geom_pointdensity.R
@@ -119,15 +119,17 @@ StatPointdensity <- ggproto("StatPointdensity", Stat,
 
                               } else if (identical(method, "kde2d")) {
 
+                                finites <- is.finite(data$x) & is.finite(data$y)
+                                ddata <- data[finites,]
                                 base.args <- list(
-                                  x = data$x,
-                                  y = data$y,
+                                  x = ddata$x,
+                                  y = ddata$y,
                                   lims = c(scales$x$dimension(), scales$y$dimension()))
                                 if (!is.element("n", names(method.args))) {
                                   method.args["n"] <- 100
                                 }
                                 if (!is.element("h", names(method.args))) {
-                                  h <- c(MASS::bandwidth.nrd(data$x), MASS::bandwidth.nrd(data$y))
+                                  h <- c(MASS::bandwidth.nrd(ddata$x), MASS::bandwidth.nrd(ddata$y))
                                   method.args$h <- h * adjust
                                 }