One hot transformation of catagorical variable in tidyverse

one-hot-encoding

library(tidyverse)

one_hot <- function(data, var) {

  var_enquo <- enquo(var)
  items <- data %>% pull(!!var_enquo)
  items_unique <- items %>% unique()
  
  out <- matrix(0, NROW(data), length(items_unique))
  colnames(out) <- items_unique
  
  for (i in items_unique) {
    out[, i] <- items == i
  }
  
  data %>%
    select(-!!var_enquo) %>%
    bind_cols(as.tibble(out))
}

one_hot(iris, Species)