R/chooseColumn.R

In elo2zero/oddsandsods: Miscellanous functions

#' Returns column names of specified class
#'
#' @param data data.frame
#' @param classes vector with classes names to select columns
#' @return symbols of columns with chosen classes
#' @examples
#' chooseColumn(data, classes=c("integer","numeric"))
#' @export
chooseColumn <- function(data, classes=c("integer","numeric")){
  classes <- paste(classes, collapse="|")
  vars <-
    lapply(data, class) %>%
    unlist %>%
    grep(classes,x=., value=T) %>%
    names() %>%
    lapply(as.symbol)
  return(vars)
}