R/get_r_est.R

In maxsal/covid19india: Pulling Clean Data from Covid19india.org

#' Helper function for pulling latest R estimates
#' @param x data set containing R estimates
#' @return Pulls 7-day trailing average R estimates and 95% confidence intervals
#' @import data.table
#' @export
#' @examples
#' \dontrun{
#' get_r_est(x = get_all_data())
#' }

get_r_est <- function(x) {

  x[order(date), .SD[ifelse(.N-6 < 1, 1, .N-6):.N], by = "place"][, .(
    r     = mean(r_est, na.rm = TRUE),
    lower = mean(r_lower, na.rm = TRUE),
    upper = mean(r_upper, na.rm = TRUE)),
    .(place)]

}