fix or #1

kassambara · Feb 17, 2017 · dd3f45b · dd3f45b
1 parent 7a110b5
commit dd3f45b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/R/ggdensity.R b/R/ggdensity.R
@@ -102,8 +102,8 @@ ggdensity <- function(data, x, y = "..density..",
       grp_name <- grp
       if(!inherits(data[, grp_name], "factor")) data[, grp_name] <- as.factor(data[, grp_name])
       df <- data.frame(grp = data[, grp_name], x = data[, x])
-      if (add == "mean") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), mean)
-      else if (add == "median") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), stats::median)
+      if (add == "mean") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), mean, na.rm = TRUE)
+      else if (add == "median") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), stats::median, na.rm = TRUE)
       names(df.m) <- c(grp_name,'x.mean')
       p <- p + .geom_exec(geom_vline, data = df.m,
                           xintercept = "x.mean", color = add.params$color,

diff --git a/R/gghistogram.R b/R/gghistogram.R
@@ -120,8 +120,8 @@ gghistogram <- function(data, x, y = "..count..",
       grp_name <- grp
       if(!inherits(data[, grp_name], "factor")) data[, grp_name] <- as.factor(data[, grp_name])
       df <- data.frame(grp = data[, grp_name], x = data[, x])
-      if (add == "mean") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), mean)
-      else if (add == "median") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), stats::median)
+      if (add == "mean") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), mean, na.rm = TRUE)
+      else if (add == "median") df.m <- stats::aggregate(df[, "x"], by = list(grp = df[, "grp"]), stats::median, na.rm = TRUE)
       names(df.m) <- c(grp_name,'x.mean')
       p <- p + .geom_exec(geom_vline, data = df.m,
                           xintercept = "x.mean", color = add.params$color,