在dplyr::filter中将字符串作为变量名传递

130

我使用mtcars数据集来阐明我的问题。

例如,我想将数据子集化为4缸汽车。我可以这样做:

mtcars %>% filter(cyl == 4)

在我的工作中,我需要将一个字符串变量作为我的列名传递。例如:

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

我还做了以下事项:

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

在这两种情况下,我得到了空数据框。


1
这个回答解决了你的问题吗?通过字符列名(在dplyr中)过滤数据框架 - camille
4个回答

144

!! 或者 UQ 会评估变量,所以 mtcars %>% filter(!!var == 4)mtcars %>% filter('cyl' == 4) 是相同的,其中条件始终为false;你可以通过在过滤函数中打印 !!var 来证明这一点:

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)
为了将 var 评估到 cyl 列,您需要先将 var 转换为 cyl 的符号,然后再将符号 cyl 评估为一列:
使用 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
# ...

或者使用baseR中的as.symbol/as.name

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

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

2
我该如何做同样的事情,但针对一个字符串列表(最终对应列名列表)? - ifreak
@Psidom,使用您提供的示例,在我的安装(使用R 3.5)上,这些方法都不起作用。 "sym"方法返回与mtcars相同大小的矩阵,但所有元素都为零。 "as.symbol"和"as.name"方法都返回“无效参数类型”错误。有什么想法吗? - DangerousDave
30
如果你需要另一个包来在dplyr中完成这个操作,那就太疯狂了。mtcars %>% filter(get(var) == 4)不好吗? - s_baldur
3
不确定使用get()是否存在问题,但是dplyr已经导入了rlang,因此它已经被安装了。 - r_alanb
8
get()更简单 - tef2128
@ifreak - 对于一个字符串列表,我不知道tidyverse的解决方案,但是你可以使用循环:for(i in 1:length(x$colWithVarNames)) x$result[i] <- x[[x$colWithVarNames[i]]][i] - user3799203

68
我认为 @snoram 的回答很优雅,并且仅依赖于 dplyr
var <- c('cyl')
mtcars %>% filter(get(var) == 4)

您还可以将其与列表一起使用。例如,您可以获取每个已过滤列的计数作为新数据集。

#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)
}

1
对于任何遇到以下问题的人:Error in rownames_to_column(mtcars, "car_name"),请导入 library(tibble)library(tidyverse)。rownames_to_column() 是 tibble 中的一个函数。 - rubengavidia0x

44

现在推荐使用 .data 代词:

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

3
如果你想使用mutate作为dplyr动词,看起来似乎这个方法行不通。 - mcstrother
2
这是唯一对我有效的答案。我不知道为什么其他答案获得这么高的赞数。谢谢! - Colin McGovern
1
@mcstrother 它是有效的。尝试mtcars%>% mutate(val = .[["cyl"]]) - Julien

9
你可以使用 eval(parse(text= 来将字符串作为变量进行评估:
mtcars %>% filter(eval(parse(text='cyl')) == 4)

enter image description here


7
我非常喜欢这个解决方案!我不明白为什么要将变量名称指定为变量(在函数中使用dyplr)会那么难。这是一个惊人而易于理解的解决方案。 - Angelo
@Angelo,因为R语言设计不佳,某些东西被指定为名称或值是由被调用者而非调用者决定的。因此,你会遇到一种常见的模糊情况,需要逐个解决。 - Stefano Borini

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接