R/multibartest.R

In alemiani/explora: Explore and Visualize Data

multibartest.var <- function(dataset, group, variables = '', legend = TRUE){
  require(ggplot2)
  require(Rmisc)

  legend <- legend

  N <- length(variables)

  df <- data.frame(
    group  = group)

  name <- list()
  p <- list()

  for(i in 1:N) {
    name[[i]] <- (variables[[i]])
    df$name <- dataset[, variables[[i]]]
    names(df)[names(df) == 'name'] <- variables[[i]]
    p[[i]] <- bartest(df[,variables[[i]]],df$group, legend = legend) + ggplot2::ggtitle(name[[i]])
  }
  Rmisc::multiplot(plotlist = p, cols = N)
}


multibartest.groups <- function(dataset, variable, groups = '', legend = TRUE){
  require(ggplot2)
  require(Rmisc)
  legend <- legend
  N <- length(groups)
  df <- data.frame(
    variable  = variable)

  name <- list()
  p <- list()

  for(i in 1:N) {
    name[[i]] <- (groups[[i]])
    df$name <- dataset[, groups[[i]]]
    names(df)[names(df) == 'name'] <- groups[[i]]
    p[[i]] <- bartest(df$variable, df[,groups[[i]]]) +
      ggplot2::ggtitle(paste(substitute(variable)[3], name[[i]], sep = '\n'))
  }
  Rmisc::multiplot(plotlist = p, cols = N)
}
# multibartest.groups(dataset = mtcars, variable = mtcars$mpg, groups = c('vs', 'am', 'gear', 'carb'))