R/rm_na_data_rows.R

In codaredistlm: Compositional Data Linear Models with Composition Redistribution

# @examples
# data(fat_data)
# fd_w_nas <- fat_data[1:10, ]
# fd_w_nas[1, "fat"] <- NA
# fd_w_nas[3, "sl"] <- NA
# fd_w_nas[5, "ed"] <- NA
# fd_w_nas
# rm_na_data_rows(fd_w_nas, c("fat", "sl", "sb", "lpa", "mvpa", "sibs", "parents", "ed"))


rm_na_data_rows <- function(dataf, cols) {

  # check we've got valid columns
  cols_exist(dataf, cols)

  dataf_thin <- dataf[, cols]

  na_sums <- apply(dataf_thin, 1, function(x) sum(is.na(x)))
  if (any(na_sums > 0)) {
    w_na_sums <- which(na_sums > 0)
    message("---\nRows ", paste(w_na_sums, collapse = ", "), " have NA values.\n---\n")
    message("Printing these rows before removing them:\n")
    # print(dataf_thin[w_na_sums, ])
    message(paste(kable(dataf_thin[w_na_sums, ]), collapse = "\n"), appendLF = TRUE)
    dataf_thin <- dataf_thin[!na_sums, ]
    message("\nDeletion of rows ", paste(w_na_sums, collapse = ", "), " successful.\n---\n")
  }

  return(dataf_thin)

}