给定元素向量,我想获得所有可能的n
长度元素子集组合的列表。例如,给定(最简单的)序列1:2
,我想获得表单的列表对象
{ {{1},{1}}, {{1},{2}}, {{2},{2}}, {{1},{1,2}}, {{2},{1,2}}, {{1,2},{1,2}} }
n=2
。
我能够使用以下内容生成所有非空子集的列表:
listOfAllSubsets <- function (s) {
n <- length(s)
unlist(lapply(1:n, function (n) {
combn(s, n, simplify=FALSE)
}), recursive=FALSE)
}
但是,我不确定从这里开始的最佳方式。基本上,我想要一个这个列表的笛卡尔积(对于n=2
)。
有什么建议吗?非迭代解决方案将是更可取的(即,没有for
循环)。
答案 0 :(得分:3)
从索引的笛卡尔积开始更容易。然后通过确保索引的元组被排序来避免重复。
combosn <- function(items,n) {
i <- seq_along(items)
idx <-do.call(expand.grid,rep(list(i),n))
idx <- idx[!apply(idx,1,is.unsorted),]
apply(idx,1,function(x) items[x])
}
ss<-listOfAllSubsets(1:2)
str(combosn(ss,2))
List of 6 $ :List of 2 ..$ : int 1 ..$ : int 1 $ :List of 2 ..$ : int 1 ..$ : int 2 $ :List of 2 ..$ : int 2 ..$ : int 2 $ :List of 2 ..$ : int 1 ..$ : int [1:2] 1 2 $ :List of 2 ..$ : int 2 ..$ : int [1:2] 1 2 $ :List of 2 ..$ : int [1:2] 1 2 ..$ : int [1:2] 1 2
或者,对于n=3
,
str(combosn(ss,3))
List of 10 $ :List of 3 ..$ : int 1 ..$ : int 1 ..$ : int 1 $ :List of 3 ..$ : int 1 ..$ : int 1 ..$ : int 2 $ :List of 3 ..$ : int 1 ..$ : int 2 ..$ : int 2 $ :List of 3 ..$ : int 2 ..$ : int 2 ..$ : int 2 $ :List of 3 ..$ : int 1 ..$ : int 1 ..$ : int [1:2] 1 2 $ :List of 3 ..$ : int 1 ..$ : int 2 ..$ : int [1:2] 1 2 $ :List of 3 ..$ : int 2 ..$ : int 2 ..$ : int [1:2] 1 2 $ :List of 3 ..$ : int 1 ..$ : int [1:2] 1 2 ..$ : int [1:2] 1 2 $ :List of 3 ..$ : int 2 ..$ : int [1:2] 1 2 ..$ : int [1:2] 1 2 $ :List of 3 ..$ : int [1:2] 1 2 ..$ : int [1:2] 1 2 ..$ : int [1:2] 1 2
答案 1 :(得分:2)
这就是我要做的事情,例如,s=1:2
:
1)为每个元素的成员资格表示具有0/1矩阵的子集。
subsets = as.matrix(do.call(expand.grid,replicate(length(s),0:1,simplify=FALSE)))
给出了
Var1 Var2
[1,] 0 0
[2,] 1 0
[3,] 0 1
[4,] 1 1
这里,第一行是空子集;第二,{1};第三,{2};第四,{1,2}。要获取子集本身,请使用mysubset = s[subsets[row,]]
,其中row
是您想要的子集行。
2)将子集对表示为矩阵的行对:
pairs <- expand.grid(Row1=1:nrow(subsets),Row2=1:nrow(subsets))
给出了
Row1 Row2
1 1 1
2 2 1
3 3 1
4 4 1
5 1 2
6 2 2
7 3 2
8 4 2
9 1 3
10 2 3
11 3 3
12 4 3
13 1 4
14 2 4
15 3 4
16 4 4
这里,第十四行对应于subsets
的第二和第四行,因此{1}&amp; {1,2}。这假设对的顺序很重要(这在采用笛卡尔积时是隐含的)。要恢复子集,请使用mypairosubsets=lapply(pairs[p,],function(r) s[subsets[r,]])
,其中p
是您想要的对的行。
将对数扩展到P(s)^n
案例(其中P(s)
是s
的权力集)看起来像
setsosets = as.matrix(do.call(expand.grid,replicate(n,1:nrow(subsets),simplify=FALSE)))
这里,每行都有一个数字向量。每个数字对应subsets
矩阵中的一行。
制作s
元素的副本可能不需要在此之后做的任何事情。但是,你可以使用lapply(1:nrow(pairs),function(p)lapply(pairs[p,],function(r) s[subsets[r,]]))
从这里开始,它的开始就像......
[[1]]
[[1]]$Row1
integer(0)
[[1]]$Row2
integer(0)
[[2]]
[[2]]$Row1
[1] 1
[[2]]$Row2
integer(0)
答案 2 :(得分:1)
allSubsets<-function(n,# size of initial set
m,# number of subsets
includeEmpty=FALSE)# should the empty set be consiered a subset?
{
# m can't exceed the number of possible subsets
if(includeEmpty)
stopifnot(m <= 2^n)
else
stopifnot(m <= 2^n-1)
# get the subsets of the initial set (of size n)
if(includeEmpty){
ll <- split(t(combn(2^n,m)),seq(choose(2^n,m)))
}else
ll <- split(t(combn(2^n-1,m)),seq(choose(2^n-1,m)))
# get the subets
subsets <- apply(do.call(expand.grid,rep(list(c(F,T)),n)),
1,which)
# remove the empty subset if desired
if(!includeEmpty)
subsets <- subsets[-1]
# covert the subsets to vector
subsets <- lapply(subsets,as.vector)
# return the list of subsets
apply(t(mapply('[',list(subsets),ll)),1,function(x)x)
}
# returns a list where each element is a list of length 2 with
# subsets of the initial set of length 4
x = allSubsets(4,2,F)