Uso di una valutazione non standard basata su tidyeval nella ricodifica nella parte destra del mutato

Prendi in considerazione una tabella in cui ogni colonna è un vettore di caratteri che può assumere molti valori - diciamo da "A" a "F".

library(tidyverse)
sample_df <- tibble(q1 = c("A", "B", "C"), q2 = c("B", "B", "A"))

Vorrei creare una funzione che prende il nome di una colonna come argomento e ricodifica quella colonna in modo tale che qualsiasi risposta "A" diventi una NA e il df venga altrimenti restituito così com'è. Il motivo per progettarlo in questo modo è adattarsi a una pipeline più ampia che esegue una serie di operazioni utilizzando una determinata colonna.

Ci sono molti modi per farlo. Ma sono interessato a capire quale sarebbe il miglior approccio idiomatico tidy_eval / tidyverse. Innanzitutto, il nome della domanda deve essere sul lato sinistro di un verbo mutato, quindi utilizziamo gli operatori !!e in modo :=appropriato. Ma allora, cosa mettere sul lato destro?

fix_question <- function(df, question) {
    df %>% mutate(!!question := recode(... something goes here...))
}

fix_question(sample_df, "q1") # should produce a tibble whose first column is (NA, "B", "C")

Il mio pensiero iniziale era che avrebbe funzionato:

df %>% mutate(!!question := recode(!!question, "A" = NA_character_))

Ma ovviamente il bang-bang all'interno della funzione restituisce semplicemente la stringa di caratteri letterali (ad esempio "q1"). Ho finito per prendere quello che sembra un percorso caotico per fare riferimento ai dati sul lato destro, usando l' [[operatore R di base e .basandomi sul costrutto da dplyr, e funziona, quindi in un certo senso ho risolto il mio problema di fondo:

df %>% mutate(!!question := recode(.[[question]], "A" = NA_character_))

Sono interessato a ricevere feedback da persone che sono molto brave in ordine sul fatto che esista un modo più idiomatico per farlo, nella speranza che vedere un esempio funzionante migliorerebbe la mia comprensione della funzione di ordine generale più in generale. qualche idea?

Qui, sul lato destro di :=, possiamo specificare symdi convertire in simbolo e quindi valutare ( !!)

fix_question <- function(df, question) {
    df %>%
       mutate(!!question := recode(!! rlang::sym(question), "A" = NA_character_))
  }

fix_question(sample_df, "q1") 
# A tibble: 3 x 2
#  q1    q2   
#  <chr> <chr>
#1 <NA>  B    
#2 B     B    
#3 C     A    

È un approccio migliore che funzionerebbe sia per l'input quotato che per quello non quotato ensym

fix_question <- function(df, question) {
    question <- ensym(question)
    df %>%
       mutate(!!question := recode(!! question, "A" = NA_character_))
  }


fix_question(sample_df, q1)
# A tibble: 3 x 2
#  q1    q2   
#  <chr> <chr>
#1 <NA>  B    
#2 B     B    
#3 C     A    

fix_question(sample_df, "q1")
# A tibble: 3 x 2
#  q1    q2   
#  <chr> <chr>
#1 <NA>  B    
#2 B     B    
#3 C     A    

Ora puoi usare il metodo "ricci ricci" se hai rlang> = 0.4.0 .

Spiegazione grazie a @ eipi10:

Questo combina il processo in due fasi di quote-quindi-non quotate in una fase, quindi {{question}}equivale a!!enquo(question)

fix_question <- function(df, question){
  df %>% mutate({{question}} := recode({{question}}, A = NA_character_))
}

fix_question(sample_df, q1)
# # A tibble: 3 x 2
#   q1    q2   
#   <chr> <chr>
# 1 NA    B    
# 2 B     B    
# 3 C     A    

Nota che a differenza ensymdell'approccio, questo non funziona con i nomi dei personaggi. Ancora peggio, fa la cosa sbagliata invece di dare solo un errore.

fix_question(sample_df, 'q1')

# # A tibble: 3 x 2
#   q1    q2   
#   <chr> <chr>
# 1 q1    B    
# 2 q1    B    
# 3 q1    A    

È possibile rendere la funzione un po 'più flessibile consentendo di inserire anche un argomento di valori ricodificati come argomento. Per esempio:

library(tidyverse)
sample_df <- tibble(q1 = c("A", "B", "C"), q2 = c("B", "B", "A"))

fix_question <- function(df, question, recode.vec) {

  df %>% mutate({{question}} := recode({{question}}, !!!recode.vec))

}

fix_question(sample_df, q1, c(A=NA_character_, B="Was B"))

  q1    q2   
1 <NA>  B    
2 Was B B    
3 C     A

Si noti che recode.vecè "non quotato" con!!! . Puoi vedere cosa sta facendo questo esempio, adattato dalla Programmazione con vignetta dplyr (cerca "splice" per vedere gli esempi pertinenti). Nota come !!!"giunge" le coppie di valori di ricodifica nella recodefunzione in modo che vengano utilizzati come ...argomento in recode.

x = c("A", "B", "C")
args = c(A=NA_character_, B="Was B")

quo(recode(x, !!!args))

<quosure>
expr: ^recode(x, A = <chr: NA>, B = "Was B")
env:  global

Se si desidera eseguire potenzialmente la funzione di ricodifica su più colonne, è possibile trasformarla in una funzione che accetta solo un nome di colonna e un vettore di ricodifica. Questo approccio sembra essere più amichevole.

fix_question <- function(question, recode.vec) {

  recode({{question}}, !!!recode.vec)

}

sample_df %>% 
  mutate_at(vars(matches("q")), list(~fix_question(., c(A=NA_character_, B="Was B"))))

  q1    q2   
1 <NA>  Was B
2 Was B Was B
3 C     <NA>

O per ricodificare una singola colonna:

sample_df %>% 
  mutate(q1 = fix_question(q1, c(A=NA_character_, B="Was B")))