Fix bug heatmap (#59)

zktuong · Feb 2, 2023 · 04cd8aa · 04cd8aa
1 parent 9dd9d0b
commit 04cd8aa
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/R/plot_cpdb_heatmap.R b/R/plot_cpdb_heatmap.R
@@ -76,13 +76,14 @@ plot_cpdb_heatmap <- function(scdata, idents, pvals, log1p_transform = FALSE, sh
   colnames(all_intr) <- intr_pairs
   all_count <- reshape2::melt(all_intr)
   if (!degs_analysis){
-    all_count <- all_count[all_count$value < alpha, ]
+    all_count$significant <- all_count$value < alpha
   } else {
-    all_count <- all_count[all_count$value == 1, ]
+    all_count$significant <- all_count$value == 1
   }
-  count1x <- all_count[, 1, drop = FALSE] %>%
-    group_by_all() %>%
-    summarise(COUNT = n()) %>%
+
+  count1x <- all_count %>%
+    group_by(Var1) %>%
+    summarise(COUNT = sum(significant)) %>%
     as.data.frame
   tmp <- lapply(count1x[, 1], function(x) strsplit(as.character(x), "\\|"))
   tmp <- lapply(tmp, function(x) x[[1]])