生成分类变量的所有可能组合并存储为向量（附mtcars示例）

需求说明 #

需要提取分类变量的所有包含2个及以上元素的子集组合，将每个组合用连字符-连接后，存储为字符向量。以mtcars数据集的gear列为例，先提取该列的唯一值，再生成所有符合要求的组合。

实现代码 #

# 提取mtcars$gear的唯一值并排序
unique_gear <- sort(unique(mtcars$gear))

# 生成所有大小为2到length(unique_gear)的子集，并用连字符连接
combs_list <- lapply(2:length(unique_gear), function(k) {
  combn(unique_gear, k, paste, collapse = "-")
})

# 将列表转换为向量
all_combi <- unlist(combs_list)

# 查看结果
all_combi

输出结果 #

运行上述代码后，all_combi会包含所有正确的多元素组合：

[1] "3-4"     "3-5"     "3-6"     "4-5"     "4-6"     "5-6"     "3-4-5"   "3-4-6"  
 [9] "3-5-6"   "4-5-6"   "3-4-5-6"

代码解释 #

• unique(mtcars$gear)：提取gear列的所有唯一取值，避免重复计算组合；sort()用于让组合按有序方式生成，结果更规整。
• lapply(2:length(unique_gear), ...)：遍历从2到唯一值数量的所有整数（即子集的元素个数），对每个个数生成对应的组合。
• combn(..., paste, collapse = "-")：combn()用于生成指定大小的所有组合，通过paste参数直接将每个组合的元素用连字符连接成字符串。
• unlist()：将lapply生成的列表转换为一维字符向量，得到最终的all_combi。

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