在用户创建的函数中对公式语法使用准引号？

Question

当我运行这段代码时：

# Create example data
df <- tibble(age=rnorm(10),
         income=rnorm(10))

make_model <- function(response_var, df){
  
  # Create formula
  form <- as.formula(response_var ~ .)
  
  # Create model
  model <- lm(form , data=df)
  
  # Return coefficients
  return(coef(model))
}

make_model(income, df)

我得到以下错误

 Error in eval(predvars, data, env) : object 'income' not found

如何使用准引用使这个函数起作用？我假设逻辑与我们调用 library(dplyr) 而不是 library("dplyr").

的逻辑相同

Answer 1

以下应该有效：

library(tidyverse)

# Your original function, modified
make_model <- function(df, column) {
  column <- enexpr(column)
  form <- as.formula(paste0(quo_text(column), " ~ ."))

  model <- lm(form, data = df)

  return(coef(model))
}

# Your original data and call
tibble(
    age = rnorm(10),
    income = rnorm(10)
  ) %>% 
  make_model(income)

Answer 2

使用blast()（将包含在 rlang 0.5.0 中）

blast <- function(expr, env = caller_env()) {
  eval_bare(enexpr(expr), env)
}

make_model <- function(data, column) {
  f <- blast(!!enexpr(column) ~ .)
  model <- lm(f, data = data)
  coef(model)
}

df <- data.frame(
  age = rnorm(10),
  income = rnorm(10)
)
make_model(df, income)
#> (Intercept)         age
#>  -0.3563103  -0.2200773

工作灵活：

blast(list(!!!1:3))
#> [[1]]
#> [1] 1
#>
#> [[2]]
#> [1] 2
#>
#> [[3]]
#> [1] 3

在用户创建的函数中对公式语法使用准引号？

Use quasiquotation for formula syntax in a user-created function?

r

metaprogramming

tidyeval

quasiquotes