test.R

raw_data <-
  read.csv("cleaned_data.csv") # Original DataFrame from csv file

getCounts <- function (features) {
  filteredList = vector()
  guest_data <- (raw_data) %>% filter(GUEST != "0") 
  guest_data <- guest_data[, colSums(guest_data != 0) > 0]
  
  
  bob_data <- (raw_data) %>% filter(GUEST != "1")
  for (val in features) {
    #Get data frames where all elements have 1 in the right column
    temp <- list(guest_data[guest_data[val] == 1, ])
    #add that to our list
    filteredList <- c(filteredList , temp)
  }
  count <- vector()
  for (val in filteredList) {
    count <- c(count, nrow(val))
  }
  
  #print(nrow(features))
  if (length(features) < 1) {
    plot <- "No data to display :("
  }
  else {
    names <- data.frame("Features" = features, "Counts" = count)
    plot <- ggplot(data = names, aes(x = Features, y = Counts, fill = Features)) + geom_bar(stat =
                                                                                              "identity", position = "dodge") + theme(legend.position="none")
  }
  
  return(plot)
}

#features <- c("BEACH", "CLIFF", "CLOUDS")
#getCounts(features)

#raw_data <- read.csv("cleaned_data.csv",stringsAsFactors = FALSE)

#guest_vs_solo <- function(features) {
#guest_data <- (raw_data) %>% filter(GUEST != "0") %>%
#   mutate(add_row())
#
#
#
# bob_data <- (raw_data) %>% filter(GUEST != "1") %>% select(-X, -EPISODE, -TITLE)
#
# }