CompData/best.R at master · ssantic/CompData · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
best <- function(state, outcome) {
## Read outcome data
	input <- read.csv("outcome-of-care-measures.csv")

## Check that state and outcome are valid
	if (state %in% input$State == FALSE) stop("invalid state")
	if (outcome %in% c("heart attack", "heart failure", "pneumonia") == FALSE) stop("ivalid outcome")


## Return hospital name in that state with lowest 30-day death rate
	state_input <- subset(input, State == state)

	if (outcome == "heart attack") {
		outcome_input <- state_input[,c('Hospital.30.Day.Death..Mortality..Rates.from.Heart.Attack','Hospital.Name')]
		NA_removed <- outcome_input[outcome_input$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Attack != 'Not Available',]
		numeric_outcome <- data.frame(as.numeric(levels(NA_removed$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Attack)[NA_removed$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Attack]), NA_removed$Hospital.Name)
		names(numeric_outcome)[1] <- "Hospital.30.Day.Death..Mortality..Rates.from.Heart.Attack"
		names(numeric_outcome)[2] <- "Hospital.Name"
		min_value <- min(numeric_outcome$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Attack)
		best_outcomes <- subset(numeric_outcome, Hospital.30.Day.Death..Mortality..Rates.from.Heart.Attack == min_value)
		sorted_names <- sort(best_outcomes$Hospital.Name, decreasing=TRUE)

	}

	if (outcome == "heart failure") {
	    outcome_input <- state_input[,c('Hospital.30.Day.Death..Mortality..Rates.from.Heart.Failure','Hospital.Name')]
		NA_removed <- outcome_input[outcome_input$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Failure != 'Not Available',]
		numeric_outcome <- data.frame(as.numeric(levels(NA_removed$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Failure)[NA_removed$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Failure]), NA_removed$Hospital.Name)
		names(numeric_outcome)[1] <- "Hospital.30.Day.Death..Mortality..Rates.from.Heart.Failure"
		names(numeric_outcome)[2] <- "Hospital.Name"
		min_value <- min(numeric_outcome$Hospital.30.Day.Death..Mortality..Rates.from.Heart.Failure)
		best_outcomes <- subset(numeric_outcome, Hospital.30.Day.Death..Mortality..Rates.from.Heart.Failure == min_value)
		sorted_names <- sort(best_outcomes$Hospital.Name, decreasing=TRUE)
	}

	if (outcome == "pneumonia") {
	    outcome_input <- state_input[,c('Hospital.30.Day.Death..Mortality..Rates.from.Pneumonia','Hospital.Name')]
		NA_removed <- outcome_input[outcome_input$Hospital.30.Day.Death..Mortality..Rates.from.Pneumonia != 'Not Available',]
		numeric_outcome <- data.frame(as.numeric(levels(NA_removed$Hospital.30.Day.Death..Mortality..Rates.from.Pneumonia)[NA_removed$Hospital.30.Day.Death..Mortality..Rates.from.Pneumonia]), NA_removed$Hospital.Name)
		names(numeric_outcome)[1] <- "Hospital.30.Day.Death..Mortality..Rates.from.Pneumonia"
		names(numeric_outcome)[2] <- "Hospital.Name"
		min_value <- min(numeric_outcome$Hospital.30.Day.Death..Mortality..Rates.from.Pneumonia)
		best_outcomes <- subset(numeric_outcome, Hospital.30.Day.Death..Mortality..Rates.from.Pneumonia == min_value)
		sorted_names <- sort(best_outcomes$Hospital.Name, decreasing=TRUE)
	}

	return(as.character(sorted_names[1]))

}