1

我正在尝试将用户输入解析为函数调用的参数（在表达式中）。好像我很接近但是！将我的论点用括号括起来，这是行不通的。我正在尝试使用用户输入重新创建以下内容：

recipe(mpg ~ cyl + hp + wt + disp, data = mtcars) %>% 
  step_log(disp, base = 10, offset = 0) %>% 
  prep()

library(rlang)
library(tidymodels)

user_dv <- "mpg"
user_idv <- c("cyl", "hp", "wt", "disp")

user_step <- "log"
user_selector <- "disp"
user_args <- "base = 10, offset = 0"

formula <- as.formula(paste(user_dv, paste(user_idv, collapse = " + "), sep = " ~ "))

rcp <- expr(recipe(!!formula,data = mtcars))

add_step <- function(x, step, selector, args){
  f <- parse_expr(paste0("step_", step))
  vars <- ensym(user_selector)
  
  args <- args %>% 
    str_replace(",", ";") %>% 
    parse_exprs()

  step_expr <- call2(f, vars, !!!args)
  
  expr(!!x %>% !!step_expr)
  
}

rcp %>% 
  add_step(user_step, user_selector, user_args) %>% 
  eval() %>% 
  prep()

我的表情最终看起来像这样：

recipe(mpg ~ cyl + hp + wt + disp, data = mtcars) %>% 
      step_log(disp, (base = 10), (offset = 0))

哪个不准备（）

4

1 回答 1

2

之后parse_exprs()，您最终会得到一个存储在未命名列表中的赋值表达式：

# [[1]]
# base = 10
#
# [[2]]
# offset = 0

同时，要获得您想要的效果!!!，它们需要是命名列表中的值：

# $base
# [1] 10
#
# $offset
# [1] 0

话虽如此，我建议改为“转发点”，因为它会导致更简单、更灵活的实现：

add_step <- function(x, step, selector, ...){
  f <- parse_expr(paste0("step_", step))
  vars <- sym(selector)
  step_expr <- call2(f, vars, ...)      # <---- forwarding the dots here

  expr(!!x %>% !!step_expr)
}

用户现在可以直接将所需的参数直接提供给add_step()：

rcp %>% add_step(user_step, user_selector, base=10, offset=0)

# recipe(mpg ~ cyl + hp + wt + disp, data = mtcars) %>% 
#    step_log(disp, base = 10, offset = 0)

或将它们存储在列表中并用于!!!您的功能：

user_args <- list(base = 10, offset=0)
rcp %>% add_step(user_step, user_selector, !!!user_args)
于 2020-10-09T01:07:35.730 回答