R/urmc_functionlist.R

In azavez/azavezHW6: azavezHW6

#Function List

#1 Make first row of matrix column headers
col_head <-function(x){
  col.names <- x[1,]
  y <- x[-1,]
  colnames(y) <- col.names
  y <- as.data.frame(y)
  return(y)
}

#2 Barplot That Compares Averages for Two Groups

compare.plot <- function(x, compare_column, group1, group2, group1.label = "Group 1", group2.label = "Group 2", ...){
  group1_sub <- data.frame()
  group1 <- as.list(group1)
  for (i in group1){
    append = x[compare_column == i,5:9]
    group1_sub <- rbind(group1_sub, append)}
  indx <- sapply(group1_sub, is.factor)
  group1_sub[indx] <- lapply(group1_sub[indx], function(x) as.numeric(as.character(x)))
  group1_means <- colMeans(group1_sub)
  group1_means_t <- data.frame(t(data.frame(group1_means)))
  group2_sub <-data.frame()
  group2 <- as.list(group2)
  for (i in group2){
    append2 = x[compare_column == i, 5:9]
    group2_sub <- rbind(group2_sub, append2)}
  indx2 <- sapply(group2_sub, is.factor)
  group2_sub[indx2] <- lapply(group2_sub[indx2], function(x) as.numeric(as.character(x)))
  group2_means <- colMeans(group2_sub)
  group2_means_t <- data.frame(t(data.frame(group2_means)))
  all_means <- rbind(group1_means_t, group2_means_t)
  row.names(all_means) <- c(group1.label, group2.label)
  all_means <- as.matrix(all_means)
  colnames(all_means) <- c("Studio", "Squash", "Gymnasium", "Cardio", "Weights")
  barplot(all_means, legend = c(group1.label, group2.label), main = "Average Fitness Center Member Count", col = c("navyblue","firebrick3"), beside = TRUE, ylab = "Avg. Number of Members", xlab = "Areas of Fitness Center")
}

#3 Creates Frequency Table from Character/Factor Vector
freq.plot <- function(x, xtitle = "x-axis", ytitle = "y-axis", mtitle = "main title",...){
  x <- as.factor(x)
  f_table <- as.data.frame(table(x))
  barplot(f_table$Freq, names.arg= f_table$x, xlab = xtitle, ylab = ytitle, main = mtitle, col = "grey", las = 3,...)
}