R/addIndicators.R

In biostat3: Utility Functions, Datasets and Extended Examples for Survival Analysis

Defines functions addIndicators

Documented in addIndicators

addIndicators <- function(data, formula, drop.intercept = TRUE) {
  mm <- stats::model.matrix(formula,data)
  if (drop.intercept && any(intercept <- "(Intercept)" %in% names(mm)))
      mm <- mm[, !intercept, drop = FALSE]
  for (name in colnames(mm))
      data[[name]] <- mm[, name]
  data
}

Try the biostat3 package in your browser