R/Impute_tibble.R

In JoshuaTian-McGill/DataCleansing: A collection of functions to cleanse data

#' Impute an all-character tibble
#'
#' @param input is a tibble containing NAs
#' @return is a NA-free tibble
#' @export

# roxygen2::roxygenize()

Impute_tibble <- function(input) {

  if(!is_tibble(input)) stop("input must be a tibble")

  chr_df <- input
  chr_df_modes <- sapply(chr_df, Mode)

  nc <- ncol(chr_df)

  for(i in 1:nc) {
    if(anyNA(chr_df[, i])){
      chr_df[, i] <- Impute_vector(chr_df[, i], chr_df_modes[i])
    }
  }

  return(chr_df)

}