R/dummies.R

Defines functions dummies

Documented in dummies

#' Convert a N-category vector to a N-dimension matrix
#' 
#' The function \code{dummies} converts a N-category vector to a N-dimension matrix
#' 
#' @param x A N-category vector 
#'
#' @return A N-dimension matrix with 0/1 values
#'
#' @examples
#' data(iris, package = "datasets")
#' dummies(iris[, 5])

dummies <- function(x) {
  l <- levels(factor(x))
  d <- Reduce(cbind, lapply(l, function(xi) (x == xi) + 0))
  dimnames(d) <- list(seq(length(x)), l)
  return(d)
}

Try the yap package in your browser

Any scripts or data that you put into this service are public.

yap documentation built on Oct. 26, 2020, 1:06 a.m.