将 ids -> items 转换为 {pairs of ids} -> items

Question

我有一个这样的data.frame：

x1 <- data.frame(id=1:3,item=c("A","B","A","B","C","D"))
x1[order(x1$item),]
  id item
1  1    A
3  3    A
2  2    B
4  1    B
5  2    C
6  3    D

我想得到：

id1=c(1,2,1,3,2,3)
id2 = c(2,1,3,1,3,2)
A=c(0,0,1,1,0,0)
B=c(1,1,0,0,0,0)
C = 0
D=0
datawanted <- data.frame(id1,id2,A,B,C,D)
  id1 id2 A B C D
1   1   2 0 1 0 0
2   2   1 0 1 0 0
3   1   3 1 0 0 0
4   3   1 1 0 0 0
5   2   3 0 0 0 0
6   3   2 0 0 0 0

如果 person1 和 person2 都有 B，那么在 datawanted 数据框中，A 列得到 1，否则得到 0。

有人可以给我一些R中的建议或功能来解决这个问题吗？

Answer 1

很酷的问题。你有一个二分图，所以关注Gabor's tutorial...

library(igraph)
g = graph_from_edgelist(as.matrix(x1))
V(g)$type = grepl("[A-Z]", V(g)$name)

对于OP想要的输出，首先我们可以提取关联矩阵：

gi = get.incidence(g)
#   A B C D
# 1 1 1 0 0
# 2 0 1 1 0
# 3 1 0 0 1

请注意（感谢@thelatemail），如果您不想使用 igraph，可以使用table(x1) 访问gi。

然后，我们看一下id的组合：

res = t(combn(nrow(gi), 2, function(x) c(
    as.integer(rownames(gi)[x]), 
    pmin( gi[x[1], ], gi[x[2], ] ) 
)))

dimnames(res) <- list( NULL, c("id1", "id2", colnames(gi)))
#      id1 id2 A B C D
# [1,]   1   2 0 1 0 0
# [2,]   1   3 1 0 0 0
# [3,]   2   3 0 0 0 0

这本质上是 OP 想要的输出。它们包含了多余的行（例如，1,2 和 2,1）。

---

使用图表的有趣原因 (ht Chris)：

V(g)$color <- ifelse(V(g)$type, "red", "light blue")
V(g)$x     <- (1:2)[ V(g)$type + 1 ]
V(g)$y     <- ave(seq_along(V(g)), V(g)$type, FUN = seq_along)
plot(g)

或者，显然这或多或少可以像

plot(g, layout = layout.bipartite(g)[,2:1])