make plot_GSEA more compatible to gsea_result data frame generated by…

… other means
Nesvilab · Dec 9, 2024 · c248ecc · c248ecc
1 parent f123d6d
commit c248ecc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/enrichment.R b/R/enrichment.R
@@ -66,6 +66,9 @@ GSEA_test <- function(se, col=NULL, database="GO Biological Process", file=NULL,
 
 #' @export
 plot_GSEA <- function(gsea_result, categroies=15) {
+  if (!"Count" %in% colnames(gsea_result)) {
+    gsea_result$Count <- str_count(gsea_result$core_enrichment, "/")
+  }
   temp <- gsea_result[order(gsea_result$NES, decreasing = T),]
   ID_selected <- temp[c(1:categroies, (dim(temp)[1]-categroies + 1):dim(temp)[1]),"ID"]
   temp <- temp[temp$ID %in% ID_selected,]