R/plot_scatter.R

In USEPA/Microcystinchla: Code, Data, and Manuscript for NLA, microcystin, Chl analysis

#' Scatterplot for microcystin-chla paper
#' 
#' Produces a scatterplot via ggplot
#' 
#' @export
#' @import ggplot2

plot_scatter<-function(df,xvar,yvar,cat=NULL,pt_col=1,...){
  options(scipen=5)
  xdf<-data.frame(xvar=df[[xvar]],yvar=df[[yvar]],pt_col=pt_col)
  if(!is.null(cat)){
    xdf<-data.frame(xdf,pt_col=df[[cat]])
  }
  xdf<-xdf[complete.cases(xdf),]
  x <- ggplot(xdf,aes(xvar,yvar))+
    geom_point(size=3,aes(color = pt_col)) +
    theme_bw()+
    theme(text = element_text(family="sans"),
          axis.title.x = element_text(family="sans",vjust = -0.5, size = 14),
          axis.title.y = element_text(family="sans",vjust = 1.5, size = 14),
          axis.text.x = element_text(family="sans",size = 12),
          axis.text.y = element_text(family="sans",size = 12)) + 
    labs(...)
    
  return(x)
}