R/myddt.R

In reaganrasmussen/GOODMATH4753repo: Houses functions for MATH5743

#' Plot and .csv files for DDT data
#'
#' @param df the used data frame
#' @param SPECIES the specified species
#'
#' @return A plot of DDT data, a .csv file, and list outputs
#' @importFrom dplyr '%>%' filter
#' @importFrom ggplot2 aes_string geom_smooth
#' ggplot geom_point ggtitle
#' @importFrom utils write.csv
#' @export
#'
#' @examples \dontrun{myddt(df = ddt, species = "CCATFISH")}
#'
myddt <- function(df, SPECIES)
{

  RIVER <- WEIGHT <- LENGTH <- NULL

  #rel freq before subsetting:
  tab = table(df$RIVER) / length(df$RIVER)


  newdf <- df%>% filter(SPECIES == {{SPECIES}})




  g <- ggplot(newdf, aes_string(x= "WEIGHT" ,y= "LENGTH")) +
    #the color according to RIVER
    geom_point(aes_string(color = "RIVER" )) +
    #Quadratic curve
    geom_smooth(formula = y~x +I(x^2), method = "lm") +
    #name as title
    ggtitle("Reagan Rasmussen")
  print(g)

  list = list(ddt = df, filtered = newdf, table = tab)
  print(list)

  #the data frame used to make the plot is df1
  #df1 needs to be written to the working directory as a csv
  #file called LvsWforSPECIES.csv

  if(SPECIES == "CCATFISH"){
    write.csv(newdf, "LvsWforCCATFISH.csv", row.names = FALSE)
  }
  if(SPECIES == "SMBUFFALO"){
    write.csv(newdf, "LvsWforSMBUFFALO.csv", row.names = FALSE)
  }
  if(SPECIES == "LMBASS"){
    write.csv(newdf, "LvsWforLMBASS.csv", row.names = FALSE)
  }



}