R/groupVars.R

In gastonstat/DiscriMiner: Tools of the Trade for Discriminant Analysis

#' Group Variances
#' 
#' Calculates the variances for each group
#' 
#' 
#' @param variables matrix or data frame with explanatory variables (may
#' contain missing values)
#' @param group vector or factor with group memberships
#' @param na.rm logical indicating whether missing values should be removed
#' @return matrix of group variances (with variables in the rows, and groups in
#' the columns)
#' @author Gaston Sanchez
#' @seealso \code{\link{groupMeans}}, \code{\link{groupStds}},
#' \code{\link{groupMedians}}, \code{\link{groupQuants}}
#' @export
#' @examples
#' 
#'   \dontrun{
#'   # dataset iris
#'   data(iris)
#' 
#'   # group variances
#'   groupVars(iris[,1:4], iris[,5])
#'   }
#' 
groupVars <-
function(variables, group, na.rm=FALSE)
{
  # Calculate variances by group
  # variables: matrix or data frame with explanatory variables
  # group: vector or factor with group memberships
  # na.rm: logical indicating whether missing values should be removed
  
  # check inputs
  verify_Xy = my_verify(variables, group, na.rm=na.rm)
  X = verify_Xy$X
  y = verify_Xy$y
  
  # how many groups
  ng = nlevels(y)
  # matrix with group variances
  Vars = matrix(0, ncol(X), ng)
  for (j in 1:ncol(X))
  {
    Vars[j,] = tapply(X[,j], y, FUN=var, na.rm=na.rm)
  }
  # add names
  if (is.null(colnames(X))) {
    rownames(Vars) = paste("X", 1:ncol(X), sep="")
  } else {
    rownames(Vars) = colnames(X)
  }
  colnames(Vars) = levels(y)
  # results
  Vars
}