如何正确生成R语言中向量列表的2元及以上元素组合？

嘿，我一眼就看出你遇到的问题啦——你的嵌套循环里，内层的i循环每次都会把d[[j]]的值覆盖掉，最后自然只剩下i取最大值（也就是3）时的组合结果，之前i=2的结果早就被冲掉了！

问题根源分析 #

你的原代码中，对于每个列表元素DF[[j]]，内层循环i从2到length(DF)（这里是3），每次执行d[[j]]=combn(DF[[j]],i,simplify=F)都会直接替换掉d[[j]]之前的内容。等循环结束，d[[j]]就只保留了最后一次循环（i=3）生成的组合，这就是为啥你看不到2元组合的原因。

正确的实现方式 #

我们可以用lapply来遍历每个列表元素，同时对每个数据框生成从2列到所有列的所有组合，再把这些组合合并到同一个列表里，避免覆盖问题。代码如下：

# 处理DF中的每个数据框，生成所有2个及以上元素的组合
d <- lapply(DF, function(df) {
  # 获取当前数据框的列数
  total_cols <- ncol(df)
  # 生成从2到total_cols的每个取值对应的组合，再合并成一个列表
  combo_list <- lapply(2:total_cols, function(m) {
    combn(df, m, simplify = FALSE)
  })
  # 把多个小列表合并成一个大列表，recursive=FALSE避免拆分成原子向量
  unlist(combo_list, recursive = FALSE)
})

代码解释 #

• lapply(DF, function(df)：逐个处理DF里的每个数据框
• 对每个数据框，先拿到总列数total_cols，然后用lapply(2:total_cols, ...)生成每个选取数量m对应的组合列表
• 最后用unlist(combo_list, recursive=FALSE)把不同m对应的组合列表合并成一个大列表，这样每个d[[j]]就包含了该数据框所有2元、3元（如果列数是3的话）的组合

拿你的示例DF来测试，处理后d[[1]]会包含4个元素：3组2列的组合（A&B、A&C、B&C），加上1组3列的组合（A&B&C），完全符合你的需求。

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