R/na_exclude.R

In choisy/sptools: Additional Functions for Spatial* Objects

#' Exclude features with NA attributes
#'
#' This function removes the features of a \code{SpatialPointsDataFrame} for
#' which there is NA is the attributes.
#'
#' @param spdf A \code{SpatialPointsDataFrame} object.
#'
#' @return A \code{SpatialPointsDataFrame} object.
#'
#' @importClassesFrom sp SpatialPointsDataFrame
#' @export
#'
#' @examples
#' library(imhen)
#'
#' stations <-  imhen::stations
#' stations <- sf::as_Spatial(stations)
#' stations@data$elevation[seq(2, 12, 2)] <- NA
#'
#' head(stations@data)
#' head(na_exclude(stations)@data)
na_exclude <- function(spdf) {
  sel <- which(rowSums(is.na(spdf@data)) > 0)
  if (length(sel) > 0) return(spdf[-sel, ])
  spdf
}