add group_split

NAMESPACE

@@ -13,6 +13,7 @@ S3method(full_join,Seurat)
 S3method(ggplot,Seurat)
 S3method(glimpse,tidyseurat)
 S3method(group_by,Seurat)
+S3method(group_split,Seurat)
 S3method(inner_join,Seurat)
 S3method(join_transcripts,Seurat)
 S3method(join_transcripts,default)
@@ -71,6 +72,7 @@ importFrom(dplyr,filter)
 importFrom(dplyr,full_join)
 importFrom(dplyr,group_by)
 importFrom(dplyr,group_by_drop_default)
+importFrom(dplyr,group_rows)
 importFrom(dplyr,inner_join)
 importFrom(dplyr,left_join)
 importFrom(dplyr,mutate)

R/dplyr_methods.R

@@ -966,3 +966,40 @@ pull.Seurat <- function(.data, var=-1, name=NULL, ...) {
         as_tibble() %>%
         dplyr::pull( var=!!var, name=!!name, ...)
 }
+
+#' @name group_split
+#' @rdname group_split
+#' @inherit dplyr::group_split
+#' 
+#' @examples
+#' data(pbmc_small)
+#' pbmc_small |> group_split(groups)
+#' 
+#' @importFrom ellipsis check_dots_used
+#' @importFrom dplyr group_by
+#' @importFrom dplyr group_rows
+#' @export
+group_split.Seurat <- function(.tbl, ..., .keep = TRUE) {
+
+  var_list <- enquos(...)
+
+  group_list <- .tbl |> 
+    as_tibble() |> 
+    dplyr::group_by(!!!var_list)
+
+  groups <- group_list |> 
+    dplyr::group_rows()
+
+  v <- vector(mode = "list", length = length(groups))
+
+  for (i in seq_along(v)) {
+    v[[i]] <- .tbl[,groups[[i]]]
+
+    if(.keep == FALSE) {
+      v[[i]] <- select(v[[i]], !(!!!var_list))
+    }
+  }
+
+  v
+
+}