R/Year.R

In nwfsc-assess/nwfscDiscardBootstrap: Northwest Fisheries Science Center Survey

#' Subset the data by year
#'
#' 
#' @template dat
#' @template colnm 
#' @param yrs  the years as a vector (must have c(yr,yr,etc)). Can be any integer values (i.e., c(2001, 2003:2007,2009,1999))
#' @template nmcol 
#' @template subset
#'
#' @export
year.fn<-function(dat, colnm, yrs, nmcol = paste(colnm, "new", sep=""), subset = F) {
  if(!(colnm %in% colnames(dat))) {
    stop("colnm ",colnm," is not a name in the dataframe\n")
  }
  z <- as.integer(yrs)
  dat[,nmcol] <- NA
  if(subset) {
     out <- dat[dat[,colnm] %in% z,]
     out[,nmcol] <- out[,colnm]
  }
  if(!subset) {
     out <- dat
     out[dat[,colnm] %in% z,nmcol] <- out[dat[,colnm] %in% z,colnm]  #put in years to new column and keep not selected as NA
  }
  return(invisible(out))
}