R/na_check.R

In aakosm/sandbox: Collects sands

#' NA check
#' 
#' The function checks and shows where are the NAs in the data frame or in a specific variable.
#' @param df data frame to be checked
#' @param column specific column name as string. Defaults to NULL
#' @export
#' @examples 
#' df <- data.frame(
#' x = c(1,2,NA),
#' y = c(4,NA,5),
#' z = c(9,7,6))
#' 
#' na_check(df)
#' na_check(df, "y") 

na_check <- function(df, column = NULL) {
  if (is.null(column)) {
    df[!complete.cases(df),]
  } else {
    df[is.na(df[[column]]),]
  }
}