Question

我正在为许多模型类型构建一个函数，需要提取用于制作模型的公式。有灵活的方法吗？例如：

x <- rnorm(10)
y <- rnorm(10)
z <- rnorm(10)
equation <- z ~ x + y
model <- lm(equation)

我需要做的是在传递模型后提取公式对象“方程式”。

Answer 1

你可以得到你想要的东西：

  model$call
# lm(formula = formula)

如果你想看看我发现了什么，请使用：

str(model)

由于你从调用环境中传递了'formula'（顺便选择不当的名字），你可能需要从你传递的对象中提取：

 eval(model$call[[2]])
# z ~ x + y

@JPMac提供了一种更紧凑的方法：formula(model)。还值得研究formula.lm函数使用的机制。名为formula的函数是通用的，您使用methods(formula)来查看已定义的S3方法。由于formula.lm方法的末尾有一个星号，你需要将它包装在`getAnywhere：

中

> getAnywhere(formula.lm)
A single object matching ‘formula.lm’ was found
It was found in the following places
  registered S3 method for formula from namespace stats
  namespace:stats
with value

function (x, ...) 
{
    form <- x$formula
    if (!is.null(form)) {
        form <- formula(x$terms)
        environment(form) <- environment(x$formula)
        form
    }
    else formula(x$terms)
}
<bytecode: 0x36ff26158>
<environment: namespace:stats>

所以它使用“$”来提取名为“formula”的列表项而不是从调用中拉出它。如果缺少$ formula项（在你的情况下），那么它会替换为formula(x$terms)，我怀疑它正在调用formula.default并查看该函数的操作似乎只是调整对象的环境。

Answer 2

如上所述，model$call将为您创建创建lm对象的调用，但如果该调用包含对象本身作为模型公式，则会获得对象名称，而不是公式。 / p>

可以在model$terms中访问评估对象（即公式本身）（以及有关如何处理它的一堆辅助信息）。无论调用lm的详细信息如何，这都应该有效。

从R中的模型中提取公式

2 个答案: