r - 使用 purrr 标记许多模型的回归样本

Question

我想用 purrr 标记多元回归模型中使用的样本功能。上图 this Q&A ，我可以按以下方式临时实现此目标:

library(dplyr)

df <- mtcars %>% 
  mutate(disp = replace(hp, c(2, 3), NA)) %>% 
  mutate(wt = replace(wt, c(3, 4, 5), NA))

s1 <- lm(mpg ~ disp, data = df)
df$samp1 <- TRUE
df$samp1[na.action(s1)] <- FALSE         

s2 <- lm(mpg ~ wt, data = df)
df$samp2 <- TRUE
df$samp2[na.action(s2)] <- FALSE

如何添加 samp1和 samp2至 df使用 purrr ?

Answer 1

我还没有到那里，但这里有一个使用自定义函数的整洁方法:

flag_use <- function(df, model, name) {
  mutate(df, {{name}} := !row_number() %in% na.action( {{model}} ))
}

df %>%
  flag_use(s1, "samp1") %>%
  flag_use(s2, "samp2")