列出给定大小的向量的所有子集

时间:2014-05-10 20:24:17

标签: r list subset enumerate

函数choose(n,k)告诉我们一组k个不同元素存在多少个大小为n的子集。假设我需要实际列出这些子集,我是如何创建它的?换句话说,我正在寻找一个接受向量x(长度为n)和数字k的函数,并返回一个向量列表,每个向量的大小为{{1} },子集为k。列表的长度当然应该是x。例如

choose(length(x),k)

1 个答案:

答案 0 :(得分:9)

修改


我意识到combn(1:3, 2, simplify = FALSE)会为您提供您正在寻找的列表结果。如果@Ramnath希望发布答案,那么这个答案将被删除。

> combn(1:3, 2, simplify = FALSE)
## [[1]]
## [1] 1 2

## [[2]]
## [1] 1 3

## [[3]]
## [1] 2 3

因此在其上使用*apply函数会使以下函数无关紧要。


<强>原始


利用@Ramnath的评论,你的功能可能是这样的:

enum.choose <- function(x, k) {
  if(k > length(x)) stop('k > length(x)')
  if(choose(length(x), k)==1){
    list(as.vector(combn(x, k)))
  } else {
    cbn <- combn(x, k)
    lapply(seq(ncol(cbn)), function(i) cbn[,i])
  }
}

测试运行:

> enum.choose(1:3, 2)
# [[1]]
# [1] 1 2
# 
# [[2]]
# [1] 1 3
# 
# [[3]]
# [1] 2 3
> enum.choose(c(1, 2, 5, 4), 3)
# [[1]]
# [1] 1 2 5
# 
# [[2]]
# [1] 1 2 4
# 
# [[3]]
# [1] 1 5 4
# 
# [[4]]
# [1] 2 5 4
> enum.choose(1:4, 4)
# [[1]]
# [1] 1 2 3 4
> enum.choose(1:5, 6)
# Error in enum.choose(1:5, 6) : k > length(x)