R/plot.R

In phillip-wong/oneway: One-Way Analysis of Variance

#' Plot one-way ANOVA
#'
#' plot.oneway creates group comparisons for a one-way ANOVA
#'
#' @param x an object of class oneway.
#' @param ... additional arguments passed to boxplot function
#' @import ggplot2
#'
#' @export
#' @return ggplot2 graph
#' @examples
#' # mileage <- oneway(mpg ~ cyl, mtcars)
#' plot(mileage)
plot.oneway <- function(x, ...){
  if(!inherits(x, "oneway")){
    stop("Must be class 'oneway'")
  }
  g <- as.character(x$anova$terms[[3]]) # "cyl"
  y <- as.character(x$anova$terms[[2]]) # "mpg"
  ggplot(x$anova$model, aes(x = factor(.data[[g]]), y = .data[[y]], fill = factor(.data[[g]]))) +
    geom_boxplot(...) +
    labs(x = g, y = y) +
    theme(legend.position = 'none')
}