R/descriptive_analysis_3.R

In HorridTom/clahrcnwlhf: Heart Failure Care Bundle Project Analysis, CLAHRC NWL

#' plot_age_dist
#'
#' @param df data frame containing the data to be plotted
#'
#' @return plot of age distribution
#' @export
#'
    plot_age_dist <- function(df, stratify = FALSE) {

      #dataframe restricted to only rows where 'new.spell' = TRUE
    df <- clahrcnwlhf::restrict_to_spells(df)

      #plot graph split into period.adte A and B, overall is also an option with a few edits. facet-wrap = produces the two different periods, coord_flip = flips the graphs
   age_plot <- ggplot2::ggplot(df, ggplot2::aes(x=AgeBand))+ggplot2::geom_bar(width= 0.7)+ggplot2::coord_flip()+ggplot2::ylab("Frequency")+ggplot2::xlab("Age Band")

   if (stratify) {
     age_plot <- age_plot +ggplot2::facet_wrap( ~ period.date) + ggplot2::ggtitle("Distribution by Age Bands Across Period Dates A and B")
   } else {
     age_plot <- age_plot + ggplot2::ggtitle("Distribution by Age Bands")
   }
   age_plot
}


#' plot_dist_table
#'
#' @param df dataframe
#' @param age_col column containing age band variable
#' @param stratify indicate whether table is split into periods or overall
#' @param strat_col name of column to split age band by
#'
#' @return a (two-way) summary table of age band distribution containing frequncies for each period.date and total sums for both periods
#' @export
#'
    plot_dist_table <- function(df, age_col = 'AgeBand', stratify = FALSE, strat_col){

      #dataframe restricted to only rows where 'new.spell' = TRUE
    df <- restrict_to_spells (df)

      #age band distribution for both periods + sum of both
    ageband_perioddate <- addmargins(table(df$AgeBand, df$period.date),2)

    ageband_perioddate #this is what I run in the command line

  }


#' test_plot_dist_table
#'
#' @param df dataframe
#' @param age_col column containing age band variable
#' @param stratify indicate whether table is split into periods or overall
#' @param strat_col name of column to split age band by
#'
#' @return a (two-way) summary table of age band distribution containing frequncies for each period.date and total sums for both periods
#' @export
#'
test_plot_dist_table <- function(df, age_col = 'AgeBand', stratify = FALSE, strat_col) {
      df <- restrict_to_spells (df)

      # return table dataset specified/split by period.date or else return result unspecified by period.date
      if (stratify) {
        ageband_perioddate <- addmargins(table(df$AgeBand, df$period.date),2)
      } else {
          ageband_perioddate <- table(df$AgeBand)
      }

      ageband_perioddate
}