Permalink
Find file
Fetching contributors…
Cannot retrieve contributors at this time
78 lines (68 sloc) 3.21 KB
library("dplyr")
library("ggplot2")
library("reshape2")
# download this file from Project Tycho:
# https://www.tycho.pitt.edu
# Steps:
# Register -> Log in -> Level 1 data ->
# "Search and retreive data" ->
# Options{ geographic level: state,
# disease outcome: incidence } ->
# Add all states -> "Click here to download results to Excel."
# (open in excel, export to CSV and then...)
measles <- read.csv("data/measles_incidence.csv", header=T,
skip=2, stringsAsFactors=F)
measles[measles == "-"] <- NA
measles[,-(1:2)] <- apply(measles[,-(1:2)], 2, as.numeric)
measles <- melt(measles, id.var=c("YEAR", "WEEK"))
colnames(measles) <- c("year", "week", "state", "cases")
# aggregate to yearly totals
mdf <- measles %>% group_by(state, year) %>%
summarise(c=if(all(is.na(cases))) NA else
sum(cases, na.rm=T))
mdf$state <- factor(mdf$state, levels=rev(levels(mdf$state)))
# edited from R-manual: converts AnyTHInG to Title Case
.simpleCap <- function(x) {
s <- strsplit(x, " ")[[1]]
paste(toupper(substring(s, 1, 1)), tolower(substring(s, 2)),
sep = "", collapse = " ")
}
levels(mdf$state) <- sapply(as.character(levels(mdf$state)),
function(i) .simpleCap(gsub("\\.", " ", i)))
# hack together a colourbar
cols <- c(colorRampPalette(c("#e7f0fa", "#c9e2f6", "#95cbee", "#0099dc",
"#4ab04a", "#ffd73e"))(10),
colorRampPalette(c("#eec73a", "#e29421",
"#e29421", "#f05336","#ce472e"),
bias=2)(90))
# plot to PDF device (might fail on some devices due to fonts)
pdf("figures/measles_incidence_heatmap_2.pdf", 8, 6)
ggplot(mdf, aes(y=state, x=year, fill=c)) +
geom_tile(colour="white", linewidth=2, width=.9, height=.9) + theme_minimal() +
scale_fill_gradientn(colours=cols, limits=c(0, 4000),
breaks=seq(0, 4e3, by=1e3),
na.value=rgb(246, 246, 246, max=255),
labels=c("0k", "1k", "2k", "3k", "4k"),
guide=guide_colourbar(ticks=T, nbin=50,
barheight=.5, label=T,
barwidth=10)) +
scale_x_continuous(expand=c(0,0), breaks=seq(1930, 2010, by=10)) +
geom_segment(x=1963, xend=1963, y=0, yend=51.5, size=.9, lineend = "round") +
labs(x="", y="", fill="") +
ggtitle("Measles") +
theme(legend.position=c(.5, -.13),
legend.direction="horizontal",
legend.text=element_text(colour="grey20"),
plot.margin=grid::unit(c(.5,.5,1.5,.5), "cm"),
axis.text.y=element_text(size=6, family="Helvetica", hjust=1),
axis.text.x=element_text(size=8),
axis.line.x=element_line(colour="grey20", size=2),
axis.ticks.y=element_blank(),
axis.ticks.x=element_line(colour="grey20"),
axis.ticks.length=grid::unit(.1, "cm"),
panel.grid=element_blank(),
title=element_text(hjust=-.07, face="bold", vjust=1, family="Helvetica"),
text=element_text(family="URWHelvetica")) +
annotate("text", label="Vaccine introduced", x=1963, y=53, vjust=1, hjust=0,
size=I(3), family="Helvetica")
dev.off()