R/rbubbly.R

In gabrielpreda/rbubbly: R version of Python package bubbly (mimick Gapminder visualization tools from Hans Rosling)

# rbubbly
#
# This package is used to plot animated and interactive bubble charts using Plotly
# Inspiration: https://github.com/AashitaK/bubbly
#


rbubbly <- function(dataset, x_column, y_column, bubble_column, color_column,
                    slider_column, text_column,
                    x_title=NA, y_title=NA, title=NA,
                    x_is_log=TRUE, y_is_log=FALSE,scale_bubble=1) {

  library(plotly)

  # set x_is_log flag
  if(x_is_log)
    xtype="log"
  else
    xtype=NA

  # set y_is_log flag
  if(y_is_log)
    ytype="log"
  else
    ytype=NA


  # Calculate scale factors
  xmin = min(x_column)
  xmax = max(x_column)
  ymin = min(y_column)
  ymax = max(y_column)
  smin = scale_bubble * sqrt(xmin*ymin)
  smax = scale_bubble * sqrt(xmax*ymax)

  p <-
    plot_ly(data = dataset,
      x = x_column,
      y = y_column,
      size = bubble_column,
      color = color_column,
      frame = slider_column,
      text = text_column,
      hoverinfo = "Country: ~text",
      type = 'scatter',
      mode = 'markers',
      marker = list(opacity = 0.5),
      sizes = c(smin, smax)
    ) %>%
    layout(
      title = title,
      xaxis = list(
        title = x_title,
        type = xtype
      ),
      yaxis = list(
        title = y_title,
        type = ytype
      )
    )
  return (p)
}