r - 从公式中提取所有唯一变量

Question

我将lm()模型的公式存储为chr变量，并且尝试将公式中的所有变量提取到向量中。

为了使其可复制，使用mtcars变量：mpg、、cyl和drat（我的公式有转换等，如示例中所示）

model_string <- "mpg ~ cyl + I(abs(0.5-drat)) + I((0.5 - drat)^2)"

我想要的输出是：

vars_used <- c("mpg", "cyl", "drat")

到目前为止，我所做的是：

library(tidyverse)
vars_used <- model_string %>% 
  str_extract_all(pattern = "\\w+") %>% 
  pluck(1) %>% 
  str_remove_all(pattern = "[0-9]")

vars_used

[1] "mpg"  "cyl"  "I"    "abs"  ""     ""     "drat" "I"    ""     ""     "drat" ""    

有没有更简单的方法来完成我想要做的事情，特别是使用 tidytext？

我仍然需要删除向量中的空字符串，并删除重复项。

标签： rregextidyversetidytext

Answer 1

基础包有一个功能：

all.vars(as.formula(model_string))
#[1] "mpg"  "cyl"  "drat"

在这种情况下，您不应使用文本处理。