144

假设我有一个响应变量和一个包含三个协变量的数据（作为一个玩具示例）：

y = c(1,4,6)
d = data.frame(x1 = c(4,-1,3), x2 = c(3,9,8), x3 = c(4,-4,-2))

我想对数据进行线性回归：

fit = lm(y ~ d$x1 + d$x2 + d$y2)

有没有办法写公式，这样我就不必写出每个单独的协变量？例如，像

fit = lm(y ~ d)

（我希望数据框中的每个变量都是协变量。）我问是因为我的数据框中实际上有 50 个变量，所以我想避免写出x1 + x2 + x3 + etc.

4

6 回答 6

236

在公式中可以使用一个特殊的标识符来表示所有变量，它就是.标识符。

y <- c(1,4,6)
d <- data.frame(y = y, x1 = c(4,-1,3), x2 = c(3,9,8), x3 = c(4,-4,-2))
mod <- lm(y ~ ., data = d)

您也可以这样做，以使用除一个以外的所有变量（在这种情况下不包括 x3）：

mod <- lm(y ~ . - x3, data = d)

从技术上讲，.表示公式中未提及的所有变量。例如

lm(y ~ x1 * x2 + ., data = d)

where.只会引用x3asx1并且x2已经在公式中。

于 2011-03-09T20:20:36.163 回答
70

稍微不同的方法是从字符串创建公式。在formula帮助页面中，您将找到以下示例：

## Create a formula for a model with a large number of variables:
xnam <- paste("x", 1:25, sep="")
fmla <- as.formula(paste("y ~ ", paste(xnam, collapse= "+")))

然后，如果您查看生成的公式，您将得到：

R> fmla
y ~ x1 + x2 + x3 + x4 + x5 + x6 + x7 + x8 + x9 + x10 + x11 + 
    x12 + x13 + x14 + x15 + x16 + x17 + x18 + x19 + x20 + x21 + 
    x22 + x23 + x24 + x25
于 2011-03-09T22:20:40.633 回答
9

是的，当然，只需将响应添加y为数据框中的第一列并调用lm()它：

d2<-data.frame(y,d)
> d2
  y x1 x2 x3
1 1  4  3  4
2 4 -1  9 -4
3 6  3  8 -2
> lm(d2)

Call:
lm(formula = d2)

Coefficients:
(Intercept)           x1           x2           x3  
    -5.6316       0.7895       1.1579           NA

此外，我关于 R 的信息指出，<-建议分配 with 超过=.

于 2011-03-09T20:11:46.433 回答
8

juba 方法的一个扩展是使用reformulate，这是一个专门为此类任务设计的函数。

## Create a formula for a model with a large number of variables:
xnam <- paste("x", 1:25, sep="")

reformulate(xnam, "y")
y ~ x1 + x2 + x3 + x4 + x5 + x6 + x7 + x8 + x9 + x10 + x11 + 
    x12 + x13 + x14 + x15 + x16 + x17 + x18 + x19 + x20 + x21 + 
    x22 + x23 + x24 + x25

对于 OP 中的示例，这里最简单的解决方案是

# add y variable to data.frame d
d <- cbind(y, d)
reformulate(names(d)[-1], names(d[1]))
y ~ x1 + x2 + x3

或者

mod <- lm(reformulate(names(d)[-1], names(d[1])), data=d)

请注意，将因变量添加到 data.frame ind <- cbind(y, d)是首选，不仅因为它允许使用reformulate，而且还因为它允许将来lm在函数中使用对象，如predict.

于 2017-04-01T21:11:06.650 回答
2

我构建了这个解决方案，reformulate不关心变量名是否有空格。

add_backticks = function(x) {
    paste0("`", x, "`")
}

x_lm_formula = function(x) {
    paste(add_backticks(x), collapse = " + ")
}

build_lm_formula = function(x, y){
    if (length(y)>1){
        stop("y needs to be just one variable")
    }
    as.formula(        
        paste0("`",y,"`", " ~ ", x_lm_formula(x))
    )
}

# Example
df <- data.frame(
    y = c(1,4,6), 
    x1 = c(4,-1,3), 
    x2 = c(3,9,8), 
    x3 = c(4,-4,-2)
    )

# Model Specification
columns = colnames(df)
y_cols = columns[1]
x_cols = columns[2:length(columns)]
formula = build_lm_formula(x_cols, y_cols)
formula
# output
# "`y` ~ `x1` + `x2` + `x3`"

# Run Model
lm(formula = formula, data = df)
# output
Call:
    lm(formula = formula, data = df)

Coefficients:
    (Intercept)           x1           x2           x3  
        -5.6316       0.7895       1.1579           NA

```

于 2017-11-15T00:53:41.927 回答
0

您可以检查包leaps，特别是regsubsets() 模型选择的功能。如文档中所述：

通过穷举搜索、向前或向后逐步或顺序替换来选择模型

于 2017-05-03T10:02:51.963 回答