R/rename.vars.R

In gdata: Various R Programming Tools for Data Manipulation

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
}