R语言中，在lm()函数公式中使用双括号表示法

Question

在lm公式中，是否可以使用rlang整洁评估操作符{{}}？

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

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) 

- user11151932

Answer 1

有几个要点需要注意。首先，{{ 通常只在由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)

- Artem Sokolov

Answer 2

lm 函数需要一个公式对象，你可以使用 reformulate/as.formula 来构建它。

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

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

- Ronak Shah