如何使用唯一 ID 使用循环代码构建子集

Question

我尝试在 data.frame df中编写带有 ID 的循环代码。 我现在所做的是构建另一个列表 dm ，其中包含来自df$ID ：

dm<-df %>% select(ID) %>% unique()
for (i in 1:length( dm$ID)){
   df_new<-df %>% filter(ID %in% dm$ID[i]) 
...

当前的代码可以做我需要的。 但是我想知道是否有另一种方法可以在不构建 dm 的情况下做到这一点？ 我想按 df 中的每个 ID 构建子集。 有什么建议吗？

Answer 1

不是循环遍历unique的“ID”和subset ，而是split更快的选项，它将根据“ID”的unique值将 data.frame 拆分为data.frame list

df_list <- split(df, df$ID)     

从这里，我们可以使用lapply或for循环

pdf(paste0(out_dir, output_date,'.pdf')) 
for(i in seq_along(df_list)) {
     ggplot(data = df_list[[i]]) + 
           ...
    } 
dev.off()

或与lapply

pdf(paste0(out_dir, output_date,'.pdf'))    
lapply(df_list, function(dat) 
        ggplot(data = dat) + 
           ...
    )
dev.off()        

---

关于创建unique “ID”的 object，更好的选择是

for(un in unique(df$ID)) {
       df_new <- df %>%
                    filter(ID == un)
       ggplot(df_new) + 
           ...
  }