Funktion mit dplyr, tidyr und ggplot

Question

Wie kann ich eine Funktion machen, die eine Spalte nimmt und diese in dplyr, tidyr und ggplot verwendet?

df <- data.frame(date_col = c(1,1,2,2,3,4,4,5,5), 
       col_a = c('a','b','a','b','a','a','b','a','b'), 
       val_col = runif(9)) 

Wie kann ich eine Funktion schreiben, einen Parameter param_col statt der fest codierten col_a

df %>% 
    group_by(date_col, col_a) %>% 
    summarise(val_col = sum(val_col)) %>% 
    complete(col_a, date_col) %>% 
    ggplot(aes(date_col, val_col, color = col_a)) + 
    geom_line() 

Die dplyr und ggplot Anrufe im Code arbeiten im Folgenden erläutert. Aber wie sollte der complete Aufruf geschrieben werden? Oder sollte complete_ verwendet werden?

Gibt es eine kanonische Methode, dies zu tun?

plot_nice_chart <- function(df, param_col) { 

    enq_param_col <- enquo(param_col) 
    str_param_col <- deparse(substitute(param_col)) 


    # aggregate data based on group_by_col, 
    # explicitly fill in NA's for missing to avoid interpolation 
    df %>% 
    group_by(!!enq_param_col, date_col) %>% 
    summarise(val_col = sum(val_col)) %>% 
    complete(<what-should-be-here?>, date_col) %>% 
    ggplot(aes_string("date_col", "val_col", color = str_param_col)) + 
     geom_line() 
} 

Answer 1

Die Entwicklungsversion von tidyr, tidyr_0.6.3.9000, verwendet jetzt tidyeval, wenn Sie also zu, dass aktualisieren möchten könnten Sie !! verwenden, wie Sie in group_by tat.

plot_nice_chart <- function(df, param_col) { 

    enq_param_col <- enquo(param_col) 
    str_param_col <- deparse(substitute(param_col)) 
    str_param_col 
    df %>% 
      group_by(!!enq_param_col, date_col) %>% 
      summarise(val_col = sum(val_col)) %>% 
      ungroup() %>% 
      complete(!!enq_param_col, date_col) %>% 
      ggplot(aes_string("date_col", "val_col", color = str_param_col)) + 
      geom_line() 
} 

Mit der aktuellen Version können Sie complete_ mit Variablen als Strings verwenden.

plot_nice_chart <- function(df, param_col) { 

    enq_param_col <- enquo(param_col) 
    str_param_col <- deparse(substitute(param_col)) 

    df %>% 
      group_by(!!enq_param_col, date_col) %>% 
      summarise(val_col = sum(val_col)) %>% 
      ungroup() %>% 
      complete_(c(str_param_col, "date_col")) %>% 
      ggplot(aes_string("date_col", "val_col", color = str_param_col)) + 
      geom_line() 
}