R/mean_impute.R

In bgravesteijn/impstep: Functions for the 5 steps of imputation.

#' Impute the median for continuous variables, and the mode for categorical variables.
#'
#' @param data dataframe of the data
#' @return Returns a plot showing the OR of the other variables for the missingness of the variable of interest
mean_impute <- function(data=NULL){
  for(i in 1:ncol(data)){
    if(sum(is.na(data[,i]))!=0){
      if(class(data[,i])=="factor"){
        mode_val <- names(which(table(data[,i])==max(table(data[,i]))))
        data[,i][is.na(data[,i])] <- mode_val
      }else{
        mean_val <- median(data[,i], na.rm=TRUE)
        data[,i][is.na(data[,i])] <- mean_val
      }
    }
  }
  return(data)
}