R/imputers.R

In sherrisherry/cleandata: To Inspect and Manipulate Data; and to Keep Track of This Process

# impute NAs in factorial columns by the mode of corresponding columns
impute_mode<-function(x,cols=colnames(x),idx=row.names(x),log = eval.parent(in_log_default)){
  if(is.null(dim(x)))stop(in_msg1)
  for(i in 1:length(cols))x[is.na(x[,cols[i]]),cols[i]]<-in_mode(x[idx,cols[i]])
  if(is.list(log))in_log2(x = x, cols = cols, log = log, method = 'Mode')
  return(x)
}

# impute NAs in numerical columns by the median of corresponding columns
impute_median<-function(x,cols=colnames(x),idx=row.names(x), log = eval.parent(in_log_default)){
  if(is.null(dim(x)))stop(in_msg1)
  for(i in 1:length(cols))x[is.na(x[,cols[i]]),cols[i]]<-stats::median(x[idx,cols[i]],na.rm = TRUE)
  if(is.list(log))in_log2(x = x, cols = cols, log = log, method = 'Median')
  return(x)
}

# impute NAs in numerical columns by the mean of corresponding columns
impute_mean<-function(x,cols=colnames(x),idx=row.names(x), log = eval.parent(in_log_default)){
  if(is.null(dim(x)))stop(in_msg1)
  for(i in 1:length(cols))x[is.na(x[,cols[i]]),cols[i]]<-mean(x[idx,cols[i]],na.rm = TRUE)
  if(is.list(log))in_log2(x = x, cols = cols, log = log, method = 'Mean')
  return(x)
}