在 R 中生成两列值的所有组合

Question

我有一个包含两列的 data.frame

> data.frame(a=c(5,4,3), b =c(1,2,4))
  a b
1 5 1
2 4 2
3 3 4

我想生成一个 data.frame 的列表，其中包含这些列值的不同组合；上面的例子总共应该有六种可能的情况（如果我错了请纠正我）：

有没有简单的函数可以做到？我不认为 expand.grid 适合我。

Answer 1

实际上expand.grid 可以在这里工作，但是不推荐，因为当你有很多行时它效率很低df（如果您有 n 行，则需要从 n**n 中提取 n! 个子集。

下面是一个使用 expand.grid

的例子

u <- do.call(expand.grid, rep(list(seq(nrow(df))), nrow(df)))
lapply(
  asplit(
    subset(
      u,
      apply(u, 1, FUN = function(x) length(unique(x))) == nrow(df)
    ), 1
  ), function(v) within(df, b <- b[v])
)

一个更有效的选择是使用包 pracma

中的 perms

library(pracma)
> lapply(asplit(perms(df$b),1),function(v) within(df,b<-v))
[[1]]
  a b
1 5 4
2 4 2
3 3 1

[[2]]
  a b
1 5 4
2 4 1
3 3 2

[[3]]
  a b
1 5 2
2 4 4
3 3 1

[[4]]
  a b
1 5 2
2 4 1
3 3 4

[[5]]
  a b
1 5 1
2 4 2
3 3 4

[[6]]
  a b
1 5 1
2 4 4
3 3 2

Answer 2

使用 combinat::permn 创建 b 值的所有可能排列，并将每个排列与 a 列绑定。

df <- data.frame(a= c(5,4,3), b = c(1,2,4))
result <- lapply(combinat::permn(df$b), function(x) data.frame(a = df$a, b = x))
result

#[[1]]
#  a b
#1 5 1
#2 4 2
#3 3 4

#[[2]]
#  a b
#1 5 1
#2 4 4
#3 3 2

#[[3]]
#  a b
#1 5 4
#2 4 1
#3 3 2

#[[4]]
#  a b
#1 5 4
#2 4 2
#3 3 1

#[[5]]
#  a b
#1 5 2
#2 4 4
#3 3 1

#[[6]]
#  a b
#1 5 2
#2 4 1
#3 3 4

在 R 中生成两列值的所有组合

Producing all combinations of two column values in R

combinations

r