R/include_variables.R

In ryoogata/myPackage:

#' ダミー変数化された data.frame の特定の文字列を含む列名を抽出
#'
#' @param DATA data.frame 学習用データ
#' @param VECTOR vector 抽出したい文字列
#'
#' @return vector 抽出された文字列
#'
#' @export
include_variables <- function(DATA, VECTOR){
  id <- NULL
  for(i in VECTOR){
    id <- c(id, grep(i, names(DATA)))
  }

  return(names(DATA[, id]))
}