Как получить geom_vline и facet_wrap из ggplot2 для работы внутри функции

Question

Я использую ggplot2 для изучения влияния различных военных операций на уровень убийств. Чтобы показать эффект, я рисую вертикальную линию, когда произошла операция, и сглаженную линию уровня убийств до и после операции.

Я написал график facet_wrap, чтобы показать это для целой группы округов. Это прекрасно работает, но при преобразовании в функцию я получаю ошибку при использовании локальной переменной для рисования вертикальной линии.

Вот пример кода:

drawTS <- function(df, dates, text) {
    p <- ggplot(df, aes(date, murders)) +
      facet_wrap(~ county, ncol = 1,
                 scale="free_y") +
      scale_x_date() +
      geom_smooth(aes(group = group), se = FALSE)
    for(i in 1:length(dates)) {
      #If it's not a global variable I get an object not found error
      temp[i] <<- dates[i]
      p <- p + geom_text(aes(x,y), label = text[i],
                  data = data.frame(x = dates[i], y = -10),
                  size = 3, hjust = 1, vjust = 0) +
           #Here's the problem
           geom_vline(xintercept=temp[i], alpha=.4)
    }
    p
}

library(ggplot2)
df <- data.frame(date = rep(seq(as.Date("2007/1/01"),
                          length=36, by='1 month'),4),
               murders = round(runif(36*4) * 100),
               county = rep(rep(factor(1:4),9),each=4),
               group = rep(c(rep(1,6), rep(2,12),rep(3,18))), each=4)
dates <- c(as.Date("2007/6/15"), as.Date("2008/6/15"))
temp <- c()
drawTS(df, dates, c("Op 1","Op 2"))

Там нет ошибки с глобальной переменной, но выглядит ужасно.

Если вместо переменной temp[i] я использую dates[i] внутри geom_vline(), я получу это:

Ошибка в NextMethod ("["): объект 'i' не найден

Если я оберну переменную dates[i] в aes(), я получу:

Ошибка в eval (expr, envir, enclos): объект 'county' не найден

Кто-нибудь знает, как это исправить?

Answer 1

Я не знаю, что является причиной ошибки, но исправление, которое я мог бы придумать, - заменить цикл for на фрейм данных, подобный этому:

date.df<-data.frame(d=dates,t=text)
p <- p + geom_text(aes(x=d,label=t),y=0,
                   data = date.df,
                   size = 3, hjust = 1, vjust = 0)
p<-p+geom_vline(aes(xintercept=d),data=date.df,alpha=.4)