R/fast_table.R

In inspectdf: Inspection, Comparison and Visualisation of Data Frames

#' @importFrom dplyr arrange
#' @importFrom dplyr bind_rows
#' @importFrom dplyr desc
#' @importFrom dplyr slice
#' @importFrom tibble tibble

fast_table <- function(v, show_na = TRUE, show_cnt = FALSE){
  # convert factors to character
  if('factor' %in% class(v)) v <- as.character(v)
  vsort  <- sort(v, method = "quick", na.last = TRUE)
  vals   <- unique(vsort)
  if(any(class(v) %in% c("integer", "numeric", "double"))){
    freq   <- count_levels_num(vsort)
  } else if(any(class(v) %in% c("logical", "character", "Date", "POSIXt"))){
    freq   <- count_levels_char(vsort)
  } else {
    freq <- NA
  }
  # recombine levels with frequencies
  tbl_freq <-  tibble(value = vals, prop = freq / length(v)) 
  # if frequencies are required, add them here
  if(show_cnt) tbl_freq$cnt <- freq
  tbl_freq <- tbl_freq %>% arrange(desc(prop))
  # if the values column is not numeric, then coerce to character
  if(!any("numeric" %in% class(tbl_freq$value))){
    tbl_freq$value <- as.character(tbl_freq$value)
  }
  if(!show_na){
    tbl_freq <- tbl_freq %>% filter(!is.na(value))
  }
  # return frequency table
  return(tbl_freq)
}