在R语言中按列分组生成含数据框的列表及缺失值处理

场景1：规整分组生成列表 #

给定数据框df：

df <- data.frame(a= rep(c(1:6), 4), b= c(rep('x',6), rep('y', 6), rep('z', 6), rep('f', 6)))

目标是生成列表lst，每个子数据框对应a的一个唯一值，包含该值与所有b的唯一值的组合。

实现代码：

# 提取a和b的唯一值集合
unique_a_vals <- unique(df$a)
unique_b_vals <- unique(df$b)

# 遍历每个a的唯一值，生成对应子数据框
lst <- lapply(unique_a_vals, function(x) {
  data.frame(v1 = rep(x, length(unique_b_vals)),
             v2 = unique_b_vals)
})

# 为列表元素命名为df1到df6
names(lst) <- paste0("df", unique_a_vals)

场景2：跳过缺失组合后分组生成列表 #

给定存在缺失a-b组合的数据框newdf：

newdf <- data.frame(a= c(1:5,rep(c(1:6), 2),1:4), b= c(rep('x',5), rep('y', 6), rep('z', 4), rep('f', 6)))

目标是生成列表newlst，每个子数据框对应a的一个唯一值，仅保留数据中实际存在的a-b组合。

实现代码：

# 先去重得到所有实际存在的a-b配对
valid_pairs <- unique(newdf)

# 按a分组，将每组转换为目标格式的子数据框
newlst <- lapply(split(valid_pairs, valid_pairs$a), function(group) {
  data.frame(v1 = group$a,
             v2 = group$b)
})

# 统一列表元素命名为df1到df6
names(newlst) <- paste0("df", names(newlst))

内容的提问来源于stack exchange，提问作者Milaa