r - 使用带公式的非标准评估

Question

我正在创建一个使用非标准评估来跟踪列的含义的包。该包在函数之间传递一个数据框，这些函数执行不同的操作并执行相同的列集。非标准评估非常适合：

my_select <- function(df, xcol, ycol) {
    new_df <- dplyr::select(df, !!xcol, !!ycol)
    new_df
}
my_select(mtcars, quo(wt), quo(mpg))

但是，我想要一个适用于公式的函数：

my_lm <- function(df, xcol, ycol) {
    new_lm <- lm(!!xcol, !!ycol, data=df)
    new_lm
}
my_lm(mtcars, quo(wt), quo(mpg)

返回Error in !xcol : invalid argument type。我尝试了, 和的各种组合quo()，但基本问题是我不知道需要什么样的对象。enquo()!!lm

Answer 1

@zlipp 给出了一个有效的答案，但这里有一个更新的版本

my_lm <- function(df, xcol, ycol) {
  xcol_e <- rlang::enquo(xcol)
  ycol_e <- rlang::enquo(ycol)
  form <- paste0(rlang::as_label(ycol_e), "~", rlang::as_label(xcol_e))
    new_lm <- lm(form, data=df)
    new_lm
}
my_lm(mtcars, wt, mpg)

使用 rlang 代替粘贴

my_lm <- function(.df, xcol, ycol) {
  form <- rlang::new_formula(rlang::ensym(ycol), rlang::ensym(xcol)) 
  new_lm <- lm(form, data=.df) 
  new_lm
}
my_lm(mtcars, wt, mpg)

Answer 2

您可以通过将公式的值粘贴在一起来创建公式，然后将公式传递给lm. 我确信有更好的方法可以做到这一点，但这是一种可行的方法：

library(rlang)

my_lm <- function(df, xcol, ycol) {
  form <- as.formula(paste(ycol, " ~ ", xcol)[2])
  my_lm <- lm(form, data=df)
  my_lm
}

my_lm(mtcars, quo(wt), quo(mpg))
#> 
#> Call:
#> lm(formula = form, data = mtcars)
#> 
#> Coefficients:
#> (Intercept)           wt  
#>      37.285       -5.344