如何在R中调整model.matrix公式，快速构建含二次项的全线性模型

我正在R中构建「全线性模型」（包含主效应、二阶交互项与二次项），希望通过model.matrix以最简方式生成所有可能的项组合。目前已实现仅包含主效应与所有二阶交互项的模型，代码示例如下：

M1<-matrix(rnorm(36),nrow=6)
colnames(M1)<-LETTERS[1:6]

X<-as.data.frame(M1[,1:5])
Y<-M1[,6]

f <- as.formula( ~ .^2)
MM<-model.matrix(f, X)[,-1]

lin_model<-lm(Y ~ MM)

请问是否有简便方法调整公式f，无需手动逐个添加即可纳入二次项？尤其当特征数量超过10个时，希望避免显式定义的方式。

解决方案 #

方法1：利用poly()函数快速生成二次项 #

通过poly()的degree=2参数自动生成所有变量的二次项，结合.指代全部变量与.^2生成二阶交互项，无需手动枚举变量：

# 构建包含主效应、二次项、二阶交互项的公式
f <- as.formula(~ . + poly(., degree=2, raw=TRUE) + .^2)
# 生成模型矩阵（自动去重）
MM <- model.matrix(f, X)[,-1]
# 拟合模型
lin_model <- lm(Y ~ MM)

• poly(., degree=2, raw=TRUE)：生成所有变量的原始二次项（非正交多项式）
• .^2：生成所有变量的二阶交互项
• model.matrix会自动处理重复的主效应，无需额外去重操作

方法2：动态生成二次项公式字符串 #

提取数据框变量名，批量生成二次项的公式片段，再与原交互项公式合并，适合需要明确控制项名称的场景：

# 获取所有特征变量名
vars <- colnames(X)
# 批量生成二次项的公式字符串（如I(A^2) + I(B^2) + ...）
quad_terms <- paste0("I(", vars, "^2)", collapse = " + ")
# 合并主效应、交互项与二次项
f <- as.formula(paste0("~ .^2 + ", quad_terms))
# 生成模型矩阵
MM <- model.matrix(f, X)[,-1]
# 拟合模型
lin_model <- lm(Y ~ MM)

这种方法生成的二次项会以I(A^2)、I(B^2)的形式命名，更直观易读。

内容的提问来源于stack exchange，提问作者JanieM