R语言如何将grep提取的变量名直接用于构建回归公式

首先需要修正原写法的一个常见坑：grep(pattern="V", x=data)直接传入数据框作为x参数无法得到目标变量名，需要传入列名向量并开启值返回：

var.to.retrieve <- grep(pattern = "V", x = names(data), value = TRUE)

以下是几种比手动拼接字符串更优雅的实现方式： #

方法1：使用R原生reformulate()函数（无额外依赖） #

reformulate是R内置专门用于动态构建公式的函数，语法非常简洁：

# 直接构建公式传入lm
fit <- lm(reformulate(termlabels = var.to.retrieve, response = "var.dep"), data = data)

如果需要去除截距项，直接在变量向量里加"0"即可：

fit <- lm(reformulate(termlabels = c(var.to.retrieve, "0"), response = "var.dep"), data = data)

方法2：直接在公式中使用正则匹配（无需提前提取变量） #

如果不需要复用提取到的变量向量，可以直接在公式内用选择器匹配变量，写法更短：

# 基础R实现，用正则匹配所有包含V的变量
fit <- lm(var.dep ~ ., data = data[, grep("V", names(data)) | names(data) == "var.dep"])

如果使用tidyverse生态，还可以直接用matches选择器：

library(dplyr)
fit <- lm(var.dep ~ matches("V"), data = data)

方法3：glue字符串插值（可读性更高） #

如果需要自定义复杂的公式结构，用glue插值比手动paste逻辑更清晰：

library(glue)
formula <- as.formula(glue("var.dep ~ {paste(var.to.retrieve, collapse = '+')}"))
fit <- lm(formula, data = data)

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