R/utils.R

In efinite/utile.visuals: Create Visuals for Publication

.tabulate_at_risk <- function(fit = NULL, times = NULL) {
  fit_summary <- summary(fit, times = times)
  vctrs::vec_cbind(
    strata = as.factor(
      if (is.null(fit$strata)) rep('All', length(times))
      else
        purrr::map_chr(
          as.character(fit_summary$strata),
          ~ strsplit(.x, '=')[[1]][2]
        )
    ),
    time = fit_summary$time,
    n.risk = fit_summary$n.risk
  )
}