Y~x+趋势或y~x+“趋势”的R公式，其中趋势/“趋势”作为指标？

Question

我有一个模型构建函数，它使用公式来定义模型。除了公式类型为y ~ x的常见回归情况外，我还想添加一些可能性，例如添加趋势分量作为解释变量，该变量将在函数内部定义。下面是一个例子：

modelx <- function(formula, data,...) {        

    mf <- mc <- match.call()                         
    mf <- mf[c(1L,  match(c("formula", "data"), names(mf), 0L))]
    formula_vars <- all.vars(formula)
    if ("trend" %in% formula_vars) {
       trend <- TRUE
       formula <- update.formula(formula, ~. - trend)
    } else trend <- FALSE        

    mf[[2L]] <- formula
    mf[[1L]] <- as.name("model.frame")
    mf$na.action <- as.name("na.pass")
    mf <- eval(mf, parent.frame())        
    y <- model.response(mf, "numeric")
    mt <- attr(mf, "terms")      
    X <- model.matrix(mt, mf)

    # y, X and possible trend component etc. are combined into the model object
   if(trend)
     X<-cbind(X,1:length(y))    #just an example
   list(y=y,X=X)
}

这里的想法是，公式的类型是y ~ x + trend，函数检查公式中是否有一个名为trend的变量，删除它并将标志trend转换为TRUE，稍后将使用它来为模型构建适当的趋势组件。

我想知道有没有更好的方法来实现这一点？这种方法的小问题是，可能存在用户想要使用的名称为trend的变量，并且它与模型的趋势组件混合在一起，另一个问题是，例如，这种类型的函数不起作用，因为变量trend不存在：

combn(c(trend,x1,x2),m=2,modelx,y=y)

如果我使用string "trend"而不是trend，问题是all.vars(formula)不能捕获字符串。

有任何建议如何处理这种类型的公式，或者任何指向包含这种可能性的公式的函数的指针吗？

Answer 1

这里是一个使用reshape2的解决方案

  unlist(reshape2:::parse_formula(y~x+'trend'))
[[1]]
y

[[2]]
x

[[3]]
[1] "trend"