在函数F内部,使用F的参数作为update()的参数

时间:2017-10-06 17:16:34

标签: r formula rlang

我希望有一个像my_lm这样的函数,如下所示:

library(rlang)

base_formula <- new_formula(lhs = quote(potato), 
                            rhs = quote(Sepal.Width + Petal.Length))

my_lm <- function(response) {
    lm(formula = update(old = base_formula, new = quote(response) ~ . ),
       data = iris)
}

my_lm(response = Sepal.Length)

但我遇到以下错误:

Error in model.frame.default(formula = update(old = base_formula, new = enquo(response) ~  : 
  object is not a matrix 

我怀疑我在滥用rlang,但我似乎无法弄清楚引用,不引用和制定的组合会解决这个问题。

编辑:所需的输出就像我跑了一样:

lm(formula = Sepal.Length ~ Sepal.Width + Petal.Length,

data = iris)

EDIT2:我还应该澄清,我真的对使用rlang通过update解决此问题的解决方案感兴趣,而不是使用paste,{{ 1}}和gsub

1 个答案:

答案 0 :(得分:2)

这是有用的东西

base_formula <- new_formula(lhs = quote(potato), 
                            rhs = quote(Sepal.Width + Petal.Length))

my_lm <- function(response) {
  newf <- new_formula(get_expr(enquo(response)), quote(.))
  lm(formula = update(old = base_formula, new = newf),
     data = iris)
}

my_lm(response = Sepal.Length)

这似乎有点混乱,因为quosures也基本上是公式,你正试图用它们制作一个常规公式。然后new_formula似乎不允许!!扩展。

如果你真的只想改变左手边,那么这样的事情可能更直接

my_lm <- function(response) {
  newf <- base_formula
  f_lhs(newf) <- get_expr(enquo(response))
  lm(formula = get_expr(newf),
     data = iris)
}