R/summarize_levels.R

In omair-a-khan/ddictR: Create a Data Dictionary for any Data Set

summarize_levels <- function(data = merged.df, variable) {
  if ("epoch" %nin% names(data)) {
    stop(paste0("The data set must contain the variable: epoch.\n\n"))
  }
  
  subset.df <- data[, c("epoch", variable)]
  subset.df <- rVMAP::clear_labels(subset.df)
  
  if (!is.factor(subset.df[[variable]])) {
    subset.df[[variable]] <- factor(subset.df[[variable]], levels = sort(unique(subset.df[[variable]])))
  }
  
  data.list <- list(
    overall = subset.df,
    epoch1  = subset(subset.df, epoch == 1),
    epoch2  = subset(subset.df, epoch == 2),
    epoch3  = subset(subset.df, epoch == 3),
    epoch4  = subset(subset.df, epoch == 4)
  )
  
  output.df <- purrr::map_dfc(
    data.list,
    function(x) {get_levels2(x[[variable]])[, c("N", "P")]}
    # function(x) { get_levels(x[[variable]])$Frequency }
  )
  
  # names(output.df) <- c("Overall", "Epoch 1", "Epoch 2", "Epoch 3", "Epoch 4")
  
  output.df <- cbind(
    Value = c(levels(subset.df[[variable]]), "Missing"),
    output.df,
    stringsAsFactors = FALSE
  )
  
  # knitr::kable(
    # output.df, 
    # align = c('l', rep('r', 5))
    # col.names = c("Value", "N", "%", "N", "%", "N", "%", "N", "%", "N", "%")
  # )
  
  return(output.df)
  
  # print(knitr::kable(output.df, "latex"))
}

# #kable(
#   format = "latex",
#   # booktabs = TRUE,
#   align = c('l', rep('r', 10)),
#   col.names = c("Value", "N", "%", "N", "%", "N", "%", "N", "%", "N", "%") #rep('', 11)
# ) #%>%
#kableExtra::add_header_above(c(" ", "Overall" = 2, "Epoch 1" = 2, "Epoch 2" = 2, "Epoch 3" = 2, "Epoch 4" = 2))