添加条件以扩展R中的网格?

时间:2019-12-23 16:04:26

标签: r combinations

是否有条件将条件/限制合并到R中的expand.grid中?

作为示例,如何防止在创建网格时出现组合c(1,1,1)c(2,2,2)?是否存在可解决许多限制的可扩展解决方案?我的网格很大,内存无法处理,例如Error: cannot allocate vector of size 32.0 Gb,因此在创建网格之后进行子设置是不可行的。

expand.grid(A = rep(1:2),
            B = rep(1:3),
            C = rep(1:4))

下面的代码不起作用:

expand.grid(A[!((A==1 & B==1 & C==1) & (A==2 & B==2 & C==2))] = rep(1:2),
            B[!((A==1 & B==1 & C==1) & (A==2 & B==2 & C==2))] = rep(1:3),
            C[!((A==1 & B==1 & C==1) & (A==2 & B==2 & C==2))] = rep(1:4))

as.data.frame(expand.grid(A = rep(1:2),
                          B = rep(1:3),
                          C = rep(1:4)))[!((A==1 & B==1 & C==1) & (A==2 & B==2 & C==2))]

谢谢!

2 个答案:

答案 0 :(得分:1)

如果我们创建数据集

d1 <-  expand.grid(A = rep(1:2),
        B = rep(1:3),
        C = rep(1:4))

可以使用rowSums检查数据是否等于第一列

d1[!!rowSums(d1 != d1[,1]),]

答案 1 :(得分:1)

只需分配expand.grid数据帧,然后运行所需的逻辑即可,您可以通过比较所有列之间的相等性来缩短逻辑(根据相等性的传递属性不需要{A==C):

d1 <-  expand.grid(A = rep(1:2),
                   B = rep(1:3),
                   C = rep(1:4))

d1 <-  with(d1, d1[!(A==B & B==C),])