R/fct_case_when.R

In dpashouwer/dpash: R/dpash is an R package with miscellaneous R functions that are useful to me

#' fct_case_whn
#'
#' @param ...
#'
#' @return
#' @export
#'
#' @examples
fct_case_when <- function(...) {
  args <- as.list(match.call())
  levels <- sapply(args[-1], function(f) f[[3]])  # extract RHS of formula
  levels <- levels[!is.na(levels)]
  factor(dplyr::case_when(...), levels=levels)
}