Pase una cadena como nombre de variable en dplyr::filter

Resuelto zesla asked hace 54 años • 4 respuestas

Estoy usando el conjunto de datos mtcars para ilustrar mi pregunta.

Por ejemplo, quiero subconjuntos de datos para automóviles de 4 cilindros. Puedo hacer:

mtcars %>% filter(cyl == 4)

En mi trabajo, necesito pasar una variable de cadena como nombre de mi columna. Por ejemplo:

var <- 'cyl'
mtcars %>% filter(var == 4)

Yo también hice:

mtcars %>% filter(!!var == 4)

En ambos casos, obtuve un marco de datos vacío.

zesla avatar Jan 01 '70 08:01 zesla
Aceptado

!!o UQevalúa la variable, por lo que mtcars %>% filter(!!var == 4)es lo mismo que mtcars %>% filter('cyl' == 4)donde la condición siempre se evalúa como falsa; Puedes probar esto imprimiendo !!varen la función de filtro:

mtcars %>% filter({ print(!!var); (!!var) == 4 })
# [1] "cyl"
#  [1] mpg  cyl  disp hp   drat wt   qsec vs   am   gear carb
# <0 rows> (or 0-length row.names)

Para evaluar varla cylcolumna, primero debe convertir vara un símbolo cyly luego evaluar el símbolo cyla una columna:

Usando rlang:

library(rlang)
var <- 'cyl'
mtcars %>% filter((!!sym(var)) == 4)

#    mpg cyl  disp  hp drat    wt  qsec vs am gear carb
#1  22.8   4 108.0  93 3.85 2.320 18.61  1  1    4    1
#2  24.4   4 146.7  62 3.69 3.190 20.00  1  0    4    2
#3  22.8   4 140.8  95 3.92 3.150 22.90  1  0    4    2
# ...

O usar as.symbol/as.namedesde baseR:

mtcars %>% filter((!!as.symbol(var)) == 4)

mtcars %>% filter((!!as.name(var)) == 4)
Psidom avatar Jan 12 '2018 04:01 Psidom

Creo que la respuesta de @snoram es elegante y depende únicamente de dplyr.

var <- c('cyl')
mtcars %>% filter(get(var) == 4)

También puedes usar esto con una lista. Por ejemplo, puede obtener un recuento de cada columna filtrada como un nuevo conjunto de datos.

#adding car name
mtcars <- rownames_to_column(mtcars, "car_name")

#name your vectors
vector <- c("vs","am","carb")

df2 <- data.frame()
for (variable in vector) {
  df1 <- mtcars %>% filter(get(variable) == 1) %>% summarise(variable = n_distinct(car_name)) %>% data.frame()

  df2<- rbind(df2,df1)
}
daszlosek avatar Dec 12 '2018 17:12 daszlosek

Ahora se recomienda utilizar .datael pronombre:

library(dplyr)

mtcars %>% filter(.data[[var]] == 4)

#                mpg cyl  disp  hp drat    wt  qsec vs am gear carb
#Datsun 710     22.8   4 108.0  93 3.85 2.320 18.61  1  1    4    1
#Merc 240D      24.4   4 146.7  62 3.69 3.190 20.00  1  0    4    2
#Merc 230       22.8   4 140.8  95 3.92 3.150 22.90  1  0    4    2
#Fiat 128       32.4   4  78.7  66 4.08 2.200 19.47  1  1    4    1
#Honda Civic    30.4   4  75.7  52 4.93 1.615 18.52  1  1    4    2
#Toyota Corolla 33.9   4  71.1  65 4.22 1.835 19.90  1  1    4    1
#Toyota Corona  21.5   4 120.1  97 3.70 2.465 20.01  1  0    3    1
#Fiat X1-9      27.3   4  79.0  66 4.08 1.935 18.90  1  1    4    1
#Porsche 914-2  26.0   4 120.3  91 4.43 2.140 16.70  0  1    5    2
#Lotus Europa   30.4   4  95.1 113 3.77 1.513 16.90  1  1    5    2
#Volvo 142E     21.4   4 121.0 109 4.11 2.780 18.60  1  1    4    2
Ronak Shah avatar Oct 30 '2020 12:10 Ronak Shah