R/which_maximum.R

In dsnavega/imputeForest: Missing value imputation model using randomForest

#' Which Maximum
#'
#' Wrapper for which.max function that can handle NA (return NA)
#'
#' @noRd
#'
#' @author David Navega
#' @param x numeric (logical, integer or double) vector or R object.
#'
#' @return the index of the maximum element.
#'
#' @seealso \code{\link{which.max}}
#'
which_maximum <- function(x) {
  if(any(is.na(x))){
    NA
  } else {
    which.max(x)
  }
}