R/e_data_convert_factor_to_numeric.R

In erikerhardt/erikmisc: Erik Erhardt's miscellaneous functions for solving complex data analysis workflows

#' Convert all factor variables to their numeric factor levels
#'
#' When saving to csv, the default data class/type for factors is the character labels.  This function replaces those with the numeric factor levels.
#'
#' @param dat data.frame or tibble
#'
#' @return dat with factor variables converted to numeric
#' @export
#'
#' @examples
#' dat <- dat_mtcars_e
#' str(dat)
#'
#' dat2 <- dat |> e_data_convert_factor_to_numeric()
#' str(dat2)
#'
#' # readr::write_csv(x = dat , file = "dat.csv"    )
#' # readr::write_csv(x = dat2, file = "dat_num.csv")
e_data_convert_factor_to_numeric <-
  function(
    dat
  ) {
  ### works for data.frame but not tibble
  # for (i_var in 1:ncol(dat)) {
  #   if("factor" %in% class(dat[,i_var])) {
  #     dat[,i_var] <- as.numeric(dat[,i_var])
  #   }
  # }

  # factor column names
  names_col_factor <-
    (sapply(dat, class) == "factor") |>
    which() |>
    names()

  # apply as.numeric to all factor column names
  dat <-
    dat |>
    dplyr::mutate(
      dplyr::across(
        .cols = tidyselect::all_of(names_col_factor)
      , .fns  = as.numeric
      )
    )

  return(dat)
}