R/transvar.R

In HallLab/clarite: CLeaning to Analysis: Reproducibility-based Interface for Traits and Exposures

#' transvar
#'
#' Transform user list of columns and transform type for each column (sqrt,log)\cr
#' Can mix transform types but can only do one transform per variable at a time
#' @param df data frame
#' @param key data frame with columns for variable and transform type
#' @return data frame with the variables transformed according to the key
#' @export
#' @examples
#' \dontrun{
#' transvar(df, key)
#' }

transvar <- function(df, key) {
  t1 <- Sys.time()
  print("Running...")

  if(is.element('ID', names(df))==FALSE){
    stop("Please add ID to dataframe as column 1")
  }

  t_key <- data.frame(t(key))
  names(t_key) <- as.character(unlist(t_key[1,]))

  for(n in names(t_key)){
    #Get the function to preform
    fun <- as.character(t_key[[n]][2])
    df[,n] <- do.call(fun, list(df[,n]))
  }

  t2 <- Sys.time()
  print(paste("Finished in", round(as.numeric(difftime(t2,t1, units="secs")), 6), "secs", sep=" "))

  return(df)

}