DiscreteCovariatesToOneHot {EpiForsk}

One-hot encode factors

Description

Convert factors in a data frame to one-hot encoding.

Usage

DiscreteCovariatesToOneHot(df, factors = dplyr::everything())

Arguments

df

A data frame, data frame extension (e.g. a tibble), or a lazy data frame (e.g. from dbplyr or dtplyr).

factors

<tidy-select> One or more unquoted expressions naming factors in df to one-hot encode.

Value

Data frame with one-hot encoded factors. One-hot encoded columns have names ⁠{fct_nm}_{lvl_nm}⁠.

Author(s)

KIJA

Examples

mtcars |>
dplyr::mutate(dplyr::across(c(2, 8:11), factor)) |>
 as.data.frame() |>
 DiscreteCovariatesToOneHot(cyl)
mtcars |>
dplyr::mutate(dplyr::across(c(2, 8:11), factor)) |>
 as.data.frame() |>
 DiscreteCovariatesToOneHot(c(2, 8:11))