R/impute.R

In the-mad-statter/washu: Helpful Functions for WashU Statistical Consultations

#' Replace missing values
#'
#' @param .data a data frame, data frame extension (e.g. a tibble), or a lazy
#' data frame (e.g. from dbplyr or dtplyr)
#' @param x name of variable to impute on
#'
#' @return an object of the same type as .data with missing values in the
#' desired column imputed by random sample from observed values
#' @export
#'
#' @seealso \code{\link[tidyimpute]{impute}}
#'
#' @examples
#' impute_bootstrap(airquality, Ozone)
impute_bootstrap <- function(.data, x) {
  orig <- .data %>% dplyr::pull({{ x }})
  boot <- sample(stats::na.omit(orig), length(orig), TRUE)
  impt <- dplyr::if_else(is.na(orig), boot, orig)
  .data %>% dplyr::mutate({{ x }} := impt)
}