0

我正在编写一个函数,该函数dplyr::filter在执行其他操作之前结合了一个步骤(我想对其进行参数化)。我想为可以覆盖的过滤条件提供一个默认参数。这排除了通过... 一次尝试传递过滤参数,如下所示:

library(rlang)
library(dplyr)
filter_and_stuff1 = function(tbl, filter_args = list(mpg > 33, gear == 4), arg3, arg4){
    as_expr = enexpr(filter_args)
    sub_tbl = filter(tbl, !!!as_expr)
    # do some other things with sub_tbl, arg3 and arg4
    sub_tbl
}

filter_and_stuff1(mtcars)
Error: Argument 2 filter condition does not evaluate to a logical vector

逗号分隔似乎产生了问题。查看 dplyr 的代码,它是通过调用内部函数来处理的,该函数 quo_reduce似乎将逗号分隔的值与&. 我不明白如何在不使用的情况下做到这一点...

TLDR:如何以编程方式将一组参数传递给包含默认表达式的 dplyr::filter?

4

2 回答 2

2

问题在于,通过您使用它的方式,enexpr还捕获了对 的调用list,而使用省略号已经将每个表达式分隔为列表的不同元素:

library(rlang)

foo <- function(x) {
    enexpr(x)
}

foo(list(a, b, c))
# list(a, b, c)

bar <- function(...) {
    enexprs(...)
}

bar(a, b, c)
# [[1]]
# a
# 
# [[2]]
# b
# 
# [[3]]
# c

为了做你想做的事,你可以使用call_args从给定的内容中提取每个表达式list

baz <- function(x) {
    as_expr <- enexpr(x)
    # expr just to show
    expr(filter(!!!call_args(as_expr)))
}

baz(list(a == 1, b < 2))
# filter(a == 1, b < 2)
于 2019-06-21T10:17:11.300 回答
1

如果我放弃多个参数filter可以用逗号分隔的要求,则可以:

filter_and_stuff2 = function(tbl, filter_args = mpg > 33 & gear == 4, arg3, arg4){
    filter(tbl, !!enexpr(filter_args))
}

> filter_and_stuff2(mtcars)
   mpg cyl disp hp drat    wt qsec vs am gear carb gear4
1 33.9   4 71.1 65 4.22 1.835 19.9  1  1    4    1  TRUE
于 2019-06-20T23:03:21.833 回答