Make the prognosis annotation mode generic

pasmopy · Jun 10, 2021 · 937331d · 937331d
1 parent b905f3b
commit 937331d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/classification/brca_heatmap.R b/classification/brca_heatmap.R
@@ -54,6 +54,15 @@ suppressWarnings(annotation$days_to_death <- as.numeric(annotation$days_to_death
 annotation[is.na(annotation)] <- 0
 
 
+#add prognosis information
+for (i in 1:nrow(annotation)){
+  if (annotation$days_to_death[i] == 0){
+    annotation$prognosis[i] <- 20
+  } else {
+    annotation$prognosis[i] <- floor(annotation$days_to_death[i]/365)
+  }
+}
+
 
 #heatmap
 dynamics_info_zscore <- dynamics_info  %>% column_to_rownames(var = "Sample") %>% scale()