R：二次回归所有因素：如何指定公式？

Question

我想使用线性、成对交互和二次项对数据集上的所有可用因子指定二次回归。举个例子，如果我有

y<-rnorm(1000)
A<-sample(1:4,1000,replace=TRUE)
B<-sample(1:2,1000,replace=TRUE)
C<-sample(1:10,1000,replace=TRUE)
Dataset<-data.frame(y,A,B,C)

如果我想以这种方式回归 y 我的所有因素，我希望有一个类似的模型

model<-lm(y~.+.^2+I(A^2)+I(B^2)+I(C^2),data=Dataset)

但我想知道是否有一个表达式可以指定所有因素的所有二次项，所以我不必指定所有I(A^2)+I(B^2)+I(C^2)。

我试过了

model<-lm(y~.+.^2+I(.^2),data=Dataset)

它不起作用。你能帮我解决这个问题吗？ 谢谢。

Answer 1

我没有看到直接的解决方案，所以我想出了一种方法来重建您的公式。这适用于任意数量的变量：

library(dplyr)
data <- mtcars
y    <- "cyl"
formula <-  paste0("I(", names(data)[names(data)!=y], "^2)+", collapse="") %>%
  paste(y, "~", .) %>%
  substr(., 1, nchar(.)-1) %>%
  as.formula
lm(data=mtcars, formula)

您可以对其进行修改以包含单个因素和交互作用。 dplyr 库在这里只是为了脚本的清晰性（%>% 运算符）