R/surveys.R

In lares: Analytics & Machine Learning Sidekick

####################################################################
#' Visualize Survey Results
#'
#' This function lets the user plot a survey's result.
#'
#' @family Visualization
#' @param answers Dataframe. Answers. Each row a different person.
#' Each column a different answer.
#' @param ignore Numeric Vector. Which columns are NOT answers?
#' @param title Character. Title for your plot
#' @param subtitle Character. Subtitle for your plot.
#' @return ggplot2 object
#' @export
plot_survey <- function(answers, ignore = 1, title = NA, subtitle = NA) {
  if (length(ignore) > 0) {
    answers <- answers[-ignore]
  }
  p <- answers %>%
    gather() %>%
    freqs(.data$key, .data$value) %>%
    ggplot(aes(
      x = .data$key, y = .data$n, fill = .data$value,
      group = .data$key, label = .data$p
    )) +
    geom_bar(stat = "identity") +
    theme_minimal() +
    geom_text(position = position_stack(vjust = .5), colour = "white", size = 2.7) +
    coord_flip() +
    labs(
      title = "Survey Results",
      subtitle = "Answers for the n question",
      x = "", y = "[%]", fill = "",
      caption = paste("Obs.:", nrow(answers))
    )
  if (!is.na(title)) {
    p <- p + labs(title = title)
  }
  if (!is.na(subtitle)) {
    p <- p + labs(subtitle = subtitle)
  }
  return(p)
}