Add a contrast to a data frame

Usage

add_contrast(
  data,
  col,
  contrast = c("anova", "sum", "treatment", "helmert", "poly", "difference"),
  levels = NULL,
  ...,
  add_cols = TRUE,
  colnames = NULL
)

Arguments

data: the data frame
col: the column to recode
contrast: the contrast to recode to
levels: the levels of the factor in order
...: arguments to pass to the contrast function (base or omit)
add_cols: whether to just add the contrast to the existing column or also to create new explicit columns in the dataset (default)
colnames: optional list of column names for the added contrasts

Value

the data frame with the recoded column and added columns (if add_cols == TRUE)

Examples

df <- sim_design(between = list(time = 1:6), plot = FALSE) %>%
   add_contrast("time", "poly")

# test all polynomial contrasts
lm(y ~ time, df) %>% broom::tidy()
#> # A tibble: 6 × 5
#>   term        estimate std.error statistic p.value
#>   <chr>          <dbl>     <dbl>     <dbl>   <dbl>
#> 1 (Intercept)   0.0417    0.0422     0.989   0.323
#> 2 time^1       -0.118     0.103     -1.14    0.253
#> 3 time^2       -0.0230    0.103     -0.222   0.824
#> 4 time^3       -0.116     0.103     -1.12    0.263
#> 5 time^4       -0.0941    0.103     -0.911   0.363
#> 6 time^5       -0.0806    0.103     -0.780   0.436

# test only the linear and quadratic contrasts
lm(y ~ `time^1` + `time^2`, df) %>% broom::tidy()
#> # A tibble: 3 × 5
#>   term        estimate std.error statistic p.value
#>   <chr>          <dbl>     <dbl>     <dbl>   <dbl>
#> 1 (Intercept)   0.0417    0.0422     0.989   0.323
#> 2 `time^1`     -0.118     0.103     -1.14    0.253
#> 3 `time^2`     -0.0230    0.103     -0.222   0.824