R/dendrogram.R

In HeeseokMoon/ggedachart: Get You to Your Desired Exploratory Data Analysis Plot Faster

#' Groups Dendrogram.
#'
#' dendrogram function will draw Dendrogram for Groups analysis.
#' @param data input data.frame
#' @param method hierarchical cluster method
#' @param value value variable
#' @param title main title
#' @param subtitle subtitle
#' @param caption caption
#' @return An object of class \code{ggplot}
#' @examples
#' hc <- hclust(dist(USArrests), "ave")  # hierarchical clustering
#'
#' dendrogram(data=USArrests)
#' dendrogram(hc)
#'
#' @import ggplot2
#' @import scales
#' @import reshape2
#' @import ggthemes
#' @import gganimate
#' @import gapminder
#' @import ggalt
#' @import ggExtra
#' @import ggcorrplot
#' @import dplyr
#' @import treemapify
#' @import ggfortify
#' @import zoo
#' @import ggdendro
#' @export
dendrogram<-function(data,
                     method ="complete",
                     title=NULL,subtitle=NULL,caption=NULL){
  df<-data
  method<-method

  if(class(df)=="hclust"){
    p<-ggdendrogram(df, rotate = TRUE, size = 2) +
      theme_fivethirtyeight() +
      theme(legend.title = element_text(face = 4,size = 10),
            legend.direction = "horizontal", legend.box = "horizontal",
            panel.grid.major.y = element_blank()) +
      labs(title=title,
           subtitle=subtitle,
           caption=caption)
  } else {
    hc <- hclust(dist(df), method=method)

    p<-ggdendrogram(hc, rotate = TRUE, size = 2)+
      theme_fivethirtyeight() +
      theme(legend.title = element_text(face = 4,size = 10),
            legend.direction = "horizontal", legend.box = "horizontal",
            panel.grid.major.y = element_blank()) +
      labs(title=title,
           subtitle=subtitle,
           caption=caption)
  }

  return(p)
}