时间戳

首页 > 解决方案 > lm（）公式中的rlang双花括号

r - lm（）公式中的rlang双花括号

问题描述

{{是否可以在 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)

标签： rrlang

解决方案

几个细微差别。首先，{{通常仅在由 tidyverse 提供支持的功能中支持。即将推出的rlang::inject()功能将允许您将该支持扩展到任意功能。

其次，{{是 的简写!!enquo()，它捕获提供给函数的表达式以及应该评估该表达式的环境。由于环境已经由 data frame 提供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)

推荐阅读