r - 在矩阵 B 中查找来自矩阵 A 的行的索引

Question

让我们考虑两个矩阵 A 和 B。A 是 B 的子集。如何在矩阵 B 中找到 A 的每一行的索引？
这是一个可复制的示例:

set.seed(30)
B <- matrix(rnorm(n =30,mean = 0), ncol=3)
A <- subset(B, B[,1] > 1)

目标是找到索引 idx在这种情况下，它给出了第 4 行和第 5 行。

Answer 1

嵌套 apply循环应该这样做。

apply(A, 1, function(a)
   which(apply(B, 1, function(b) all(b==a)))
 )

# [1] 4 5

或者，使用 colSums

apply(A, 1, function(a) 
    which(colSums(t(B) == a) == ncol(B)))
# [1] 4 5