如何使用purrr从线性回归拟合对象列表中提取/保留指定元素?
解决线性回归对象列表保留指定元素的问题
你的代码报错是因为map(keep, names(.) %in% c("residuals", "fitted.values"))里的.指向的是整个回归对象列表,而非单个lm拟合结果,导致筛选条件的长度和每个lm对象的元素长度不匹配,触发了length(.p) == length(.x)的校验错误。
下面是两种可行的解决方法:
方法1:修正keep的调用方式
在map中使用匿名函数,确保每个lm对象单独被处理:
c('hp', 'wt', 'disp') %>% paste('mpg ~', .) %>% map(as.formula) %>% map(lm, data = mtcars) %>% map(~ keep(.x, names(.x) %in% c("residuals", "fitted.values")))
方法2:直接对列表取子集(更简洁高效)
lm拟合对象本质是列表,直接通过索引提取目标元素即可,无需调用keep:
c('hp', 'wt', 'disp') %>% paste('mpg ~', .) %>% map(as.formula) %>% map(lm, data = mtcars) %>% map(~ .x[c("residuals", "fitted.values")])
内容的提问来源于stack exchange,提问作者J. Doe
相关产品推荐
相关产品推荐

