R/contact_countries.R

In contactdata: Social Contact Matrices for 177 Countries

#' Get a data.frame (in long format) of contact data for multiple countries
#'
#' @param countries A character string or a vector of character containing the
#'    names of the countries for which to return contact data
#' @param ... Arguments passed to [contact_matrix()]
#'
#' @return A data.frame (in long format) with 4 columns:
#'   * `country`: the country name
#'   * `age_from`: the age group of individual
#'   * `age_to`: the age group of contact
#'   * `contact`: the intensity of contact
#'
#' @examples
#' contact_df_countries(c("Austria", "Belgium"), location = "all")
#'
#' @export
#'
#' @importFrom stats reshape
#'
#' @inherit list_countries references

contact_df_countries <- function(
  countries,
  ...
) {

  res <- lapply(countries, contact_matrix, ...)

  res <- lapply(seq_along(res), function(i) {
    out <- reshape(
      as.data.frame(res[[i]]),
      idvar = "age_from",
      ids = row.names(res[[i]]),
      times = colnames(res[[i]]),
      timevar = "age_to",
      varying = list(colnames(res[[i]])),
      direction = "long"
    )
    rownames(out) <- NULL
    colnames(out)[2] <- "contact"
    out$country <- countries[[i]]
    out <- out[, c(4, 3, 1, 2)]
    return(out)
  })

  do.call(rbind.data.frame, res)
}