R/limits_to_agegroups.r

In socialmixr: Social Mixing Matrices for Infectious Disease Modelling

#' Convert lower age limits to age groups.
#'
#' Mostly used for plot labelling
#' @param x age limits to transform
#' @param limits lower age limits; if not given, will use all limits in `x`
#' @return Age groups (limits separated by dashes)
#' @examples
#' limits_to_agegroups(c(0, 5, 10))
#' @export
limits_to_agegroups <- function(x, limits = sort(unique(x))) {
  limits <- limits[!is.na(limits)]
  agegroups <- if (length(limits) > 1) {
    vapply(seq(1, length(limits) - 1), function(y) {
      if ((limits[y + 1] - 1) > limits[y]) {
        paste(limits[y], limits[y + 1] - 1, sep = "-")
      } else {
        paste(limits[y])
      }
    }, "")
  } else {
    NULL
  }
  agegroups <- c(agegroups, paste0(limits[length(limits)], "+"))
  agegroups <- factor(agegroups, levels = agegroups, ordered = TRUE)
  names(agegroups) <- limits
  return(unname(agegroups[as.character(x)]))
}