R/filtering_functions.R

Defines functions .is_categorical_col .select_non_id_columns

Documented in .is_categorical_col .select_non_id_columns

#' Checks column data type for either character or factor (a categorical variable)
#'
#' @param col a column of data (vector)
#' @return Boolean
#' @export
#' @example
#' mtcars %>%
#'   select_if(.is_categorical_col)
.is_categorical_col <- function(col) {
  is.character(col) | is.factor(col)
}

#' In some cases, it is important to exclude ID variables from datasets. .select_non_id_columns()
#'   takes a tibble and selects only column names which do not mention as being IDs. This is
#'   useful when an ID column is interpreted as a character variable and not numeric.
#'
#' @param data a tibble
#' @export
#' @return a tibble
.select_non_id_columns <- function(data) {
  data %>%
    select(-contains("id"))
}
HSPS-DataScience/HSPSUtils documentation built on May 24, 2019, 7:37 a.m.