group by .imp if group=NULL

amices · Dec 14, 2023 · 3de3086 · 3de3086
1 parent 6a69f13
commit 3de3086
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/R/ggmice.R b/R/ggmice.R
@@ -111,6 +111,10 @@ ggmice <- function(data = NULL,
     )
     mice_mapping <-
       utils::modifyList(mapping, ggplot2::aes(colour = .where))
+    if (is.null(mapping$group)) {
+      mice_mapping <-
+        utils::modifyList(mice_mapping, ggplot2::aes(group = .imp))
+    }
     mice_colors <-
       c("observed" = "#006CC2B3",
         "imputed" = "#B61A51B3")