是否可以{{在 lm 公式中使用 rlang tidy 求值运算符?
我知道您可以使用双花括号来定义一个通用函数,如下所示:
my_scatter <- function(df, xvar, yvar) {
ggplot(df) +
geom_point(aes(x = {{xvar}}, y = {{yvar}}))
}
my_scatter(mpg, cty, hwy)
Run Code Online (Sandbox Code Playgroud)
但我想知道是否有一种方法可以在公式中进行类似的调用,例如在 lm() 内部:
my_lm <- function(df, yvar, xvar) {
lm({{yvar}} ~ {{xvar}} , data = df)
}
my_lm(mpg, cty, hwy)
Run Code Online (Sandbox Code Playgroud)
有一些细微差别。首先,{{通常仅在 tidyverse 提供的功能中受支持。即将推出的rlang::inject()功能将允许您将该支持扩展到任意功能。
其次,{{是 的简写!!enquo(),它捕获提供给函数的表达式以及应评估该表达式的环境。由于环境已经由数据框提供df,因此这里使用的更好的动词是ensym(),它仅捕获符号。
以下适用于 rlang 0.4.10:
my_lm <- function(df, yvar, xvar) {
ysym <- rlang::ensym(yvar)
xsym <- rlang::ensym(xvar)
rlang::inject( lm(!!ysym ~ !!xsym, data=df) )
}
my_lm(mpg, cty, hwy)
Run Code Online (Sandbox Code Playgroud)