R/get_first.R

In David-Statistics/djmisc: Generally Useful Functions for R & Personal Finance

#' First `n` elements
#'
#' Simple helper function to get the first `n` elements of a vector
#'
#' @param x vector to get the first `n` elements of
#' @param n number of elements to return
#' @export
get_first <- function(x, n = 1) {
  if(length(x) == 0) {
    warning('vector is empty')
    return(NULL)
  }
  if(length(x) <= n) {
    warning('n >= length(x). Returning full vector')
    return(x)
  }
  return(x[1:n])
}