R/rename.vars.R

In gdata: Various R Programming Tools for Data Manipulation

# $Id: rename.vars.R 2072 2016-02-03 20:00:57Z warnes $

rename.vars <- function(data,from='',to='',info=TRUE) {

   dsn <- deparse(substitute(data))
   dfn <- names(data)

   if ( length(from) != length(to)) {
     cat('--------- from and to not same length ---------\n')
     stop()
   }

   if (length(dfn) < length(to)) {
     cat('--------- too many new names ---------\n')
     stop()
   }

   chng <- match(from,dfn)

   frm.in <- from %in% dfn
   if (!all(frm.in) ) {
     cat('---------- some of the from names not found in',dsn,'\n')
     stop()
   }

   if (length(to) != length(unique(to))) {
     cat('---------- New names not unique\n')
     stop()
   }

   dfn.new <- dfn
   dfn.new[chng] <- to
   if (info) cat('\nChanging in',dsn)
   tmp <- rbind(from,to)
   dimnames(tmp)[[1]] <- c('From:','To:')
   dimnames(tmp)[[2]] <- rep('',length(from))
   if (info)
     {
       print(tmp,quote=FALSE)
       cat("\n")
     }
   names(data) <- dfn.new
   data
}


remove.vars <- function( data, names, info=TRUE)
{
  dsn <- deparse(substitute(data))
  if (info) cat('\nChanging in',dsn, "\n")
  
  
  flag <- names %in% colnames(data)
  if(any(!flag))
    warning("Variable(s) not found: ", paste(names[!flag], collapse=", ") )
  if(any(flag))
  {
    if(info) cat("Dropping variables:", paste(names[flag], collapse=", "), "\n\n")
    for(var in names[flag])
      data[[var]] <- NULL
  }
  data
}