R: Vars - 用于循环/函数/应用的列列表

Question

我正在尝试执行以下代码：

vars <- c("var1", "var2", "var3") #These corresponds to the names of the columns in my DF

Test <- lapply(DF[, vars], function(x) lm(x~var_x)) 

我希望能够使用分配的值“vars”，其中包括所有我想要的列（变量），这样我就不必在我的代码中一次提及每一列。但是下面的代码给了我这个错误：

 Error in `[.data.frame`(DF, , c(vars)) : undefined columns selected

我应该如何应用“vars”才能使其按需要工作？

Answer 1

这里，lapply 的输出是该列的值。可能我们需要传递列名并创建公式

lapply(vars,  function(x) lm(as.formula(paste0(x "~ var_x")), data = DF))

或者另一个选项是reformulate 来创建formula

lapply(vars, function(x) lm(reformulate("var_x", response = x), data = DF))

---

为了在单个表中获得输出，我们可以使用tidy 或glance from broom

library(purrr)
library(dplyr)
library(broom)
map_dfr(setNames(vars, vars) ~ 
            lm(reformulate("var_x", response = .x), data = DF) %>%
            tidy, .id = 'grp')