lm() 公式中的 rlang 双花括号

Question

是否可以在 lm 公式中使用 rlang tidy 求值运算符 {{？

我知道您可以使用双花括号来定义通用函数，例如：

my_scatter <- function(df, xvar, yvar) {
     ggplot(df) +
          geom_point(aes(x = {{xvar}}, y = {{yvar}}))
}

my_scatter(mpg, cty, hwy)

但我想知道是否有一种方法可以在公式中进行类似的调用，例如在 lm():

my_lm <- function(df, yvar, xvar) {
     lm({{yvar}} ~ {{xvar}} , data = df)
}

my_lm(mpg, cty, hwy)

Answer 1

lm 需要一个可以使用 reformulate/as.formula.

构建的公式对象

my_lm <- function(df, yvar, xvar) {
  lm(reformulate(xvar, yvar), data = df)
}

my_lm(mpg, 'cty', 'hwy')

Answer 2

一些细微差别。首先，{{ 通常只在由 tidyverse 提供支持的函数中受支持。 The upcoming rlang::inject() function 将允许您将该支持扩展到任意函数。

其次，{{ 是 shorthand for !!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)

lm() 公式中的 rlang 双花括号

rlang double curly braces within lm() formula

r

rlang