R/setSampleGroups.R

In nicohuttmann/htmnanalysis: Analysis of Omics Data

#' Names a group vector with groups to use
#'
#' @param group Group membership vector
#' @param n Number of groups
#'
#' @return Named group vector
#' @export
#'
#'
setSampleGroups <- function(group, n) {

  if(!hasArg(n)) {
    n <- as.numeric(readline("Number of groups: "))
  }

  message("Groups: ")

  printn(unique(group))

  group2 <- group

  group2[] <- 0

  for(i in 1:n) {
    group2[group == unique(group)[as.numeric(readline(paste("Group ", i, ": ", sep = "")))]] <- i
  }

  group2

}