R/fct_get_variables_names.R

In etiennebacher/mobilier: Build Stargazer Tables More Easily

#' Obtain variable names
#'
#' @return A vector containing the unique variables names in all regressions
#'
#' @examples NULL
get_variables_names <- function(){
  # Obtain the list of objects supported by stargazer
  list_regressions <- get_data()

  # For each regression, store variables names in a part of a list
  variables_regressions <- list()
  for (i in 1:length(list_regressions)){
    variables_regressions[[i]] <- names(list_regressions[[i]]$coefficients)
  }
  # Remove duplicates of names and store it in a vector
  all_variables <- unique(unlist(variables_regressions))
  # Rename "(Intercept)" as "Constant" since it is the name given by stargazer
  all_variables[all_variables == "(Intercept)"] <- "Constant"

  # The default behavior of stargazer is to put "Constant" at the bottom. Therefore,
  # it must be in last position in this vector of names.
  num_variables <- length(all_variables)
  all_variables <- all_variables[-1]
  all_variables[num_variables] <- "Constant"

  return(all_variables)
}