Использование ggplot в функции с фасеткой и несколькими геометрами

Question

Я пытаюсь написать функцию, которая использует ggplot, но позволяет пользователю специфицировать несколько переменных графика. Однако мне не удается заставить его работать как функция (получение сообщения об ошибке: см. Ниже).

Ниже приведен небольшой набор данных и рабочая реализация, а также моя попытка выполнить функцию и связанную с ней ошибку. Я уверен, что это связано с нестандартной оценкой (NSE), но я не уверен, как обойти это, используя мое использование фильтра внутри функции, и мои различные попытки были напрасны.

library(dplyr) 
library(ggplot2) 

df<-data.frame(Date=c(seq(1:50),seq(1:50)), SRI=runif(100,-2,2), SITE=(c(rep("A",50), rep("B", 50)))) 

ggplot() + 
geom_linerange(aes(x = Date, ymin = 0, ymax = SRI), colour = I('blue'), data = filter(df, SRI>0)) + 
geom_linerange(aes(x = Date, ymin = SRI, ymax = 0), colour = I('red'), data = filter(df, SRI<=0)) + 
facet_wrap(~SITE) + 
labs(x = 'Date', y = "yvar", title = "Plot title") 

Вышеуказанные работы, но когда реализован в виде функции:

plot_fun <- function(df, x, y, ylab="y-lab", plot_title="Title", facets) { 
ggplot() + 
geom_linerange(aes(x = x, ymin = 0, ymax = y), colour = I('blue'), data = filter(df, y > 0)) + 
geom_linerange(aes(x = x, ymin = y, ymax = 0), colour = I('red'), data = filter(df, y <= 0)) + 
facet_wrap(~ facets) + 
labs(x = 'Date', y = ylab, title = plot_title) 
return(p) 
} 

plot_fun(df, x="Date", y="SRI", ylab="y-lab", plot_title="Title", facets="SITE") 

я получаю следующее «Ошибка: Эстетика должна быть либо длина 1 или же, как данные (1): х, ymin, max ".

Я пробовал различные подходы, используя as_string и filter_, но все они не увенчались успехом.

Любая помощь очень ценится.

С уважением

Ник

Answer 1

Вам нужно переключиться на aes_string, как вы ожидали, и изменить свой facet_wrap код либо взять facets аргумент в качестве формулы или удалить тильды как в answers to this question. Вам также необходимо переключиться на использование filter_, который может использоваться вместе с interp из пакета lazyeval.

library(lazyeval) 

Вот ваша функция с изменениями, я изложил и в результате заговора:

plot_fun <- function(df, x, y, ylab = "y-lab", plot_title = "Title", facets) { 
    ggplot() + 
     geom_linerange(aes_string(x = x, ymin = 0, ymax = y), colour = I('blue'), 
        data = filter_(df, interp(~var > 0, var = as.name(y)))) + 
     geom_linerange(aes_string(x = x, ymin = y, ymax = 0), colour = I('red'), 
        data = filter_(df, interp(~var <= 0, var = as.name(y)))) + 
     facet_wrap(facets) + 
     labs(x = 'Date', y = ylab, title = plot_title) 
} 

plot_fun(df, x="Date", y="SRI", facets="SITE")