如何得到两个矩阵的交集？

Question

# These are the two matrices that I would like to subset based on identical
# entries within entire rows.
mata <- matrix(c("A", "B", "C", "F", "D", "E", "F", "G"), 
               nrow = 4, ncol = 2,
               dimnames = list(c(), c("A", "B")))
mata

##      A   B  
## [1,] "A" "D"
## [2,] "B" "E"
## [3,] "C" "F"
## [4,] "F" "G"

matb <- matrix(c("B", "A", "C", "F", "M", "D", "D", "H", "G", "X"), 
               nrow = 5, ncol = 2,
               dimnames = list(c(), c("A", "B")))
matb

##      A   B  
## [1,] "B" "D"
## [2,] "A" "D"
## [3,] "C" "H"
## [4,] "F" "G"
## [5,] "M" "X"

如果这两个矩阵不是无序的，长度相同，那么下面的代码应该可以工作，并且是有效的。

mata[rowMeans(mata == matb) == 1, ]

我的一个老练的解决方案是将我要用于匹配的每个矩阵的各个列串联起来。在本例中，我将使用所有列。

mata <- cbind(mata, C = paste0(mata[, "A"], "_", mata[, "B"]))
matb <- cbind(matb, C = paste0(matb[, "A"], "_", matb[, "B"]))
mata[mata[, "C"] %in% matb[, "C"], colnames(mata) != "C"]

##      A   B  
## [1,] "A" "D"
## [2,] "F" "G"

这就是我正在寻找的结果，但我想知道是否有更优雅的东西，比如%中的向量函数%。

该解决方案应适用于矩阵长度不一定相等的一般情况。




Answer 1

如果加载dplyr，则添加intersect.data.frame：

library(dplyr)
options(stringsAsFactors=FALSE)
dfa <- as.data.frame(mata)
dfb <- as.data.frame(matb)
intersect(dfa,dfb)

#   A B
# 1 A D
# 2 F G

类似地，Union，setEqual（测试集相等）和setdiff（set减）都是可用的。

在一边数据的每一行。frame对应于一个观察值，所以讨论两组观察值（两个data.frames）的相交是有意义的。然而，对于矩阵来说，这真的没有意义。这就是为什么如果你想继续使用矩阵，这个操作需要像OP的解决方案和@RHertel的（在幕后强制使用data.frame）这样的黑客。

Answer 2

您可以使用函数并合并（）：

> merge(mata,matb)
  A B
1 A D
2 F G