r - 检查矩阵行是否等于 R 中的向量，向量化

Question

我很惊讶这个问题没有被问到，也许答案会澄清原因。我想将矩阵的行与向量进行比较，并返回行是否无处不在的向量。请参阅下面的示例。我想要一个矢量化的解决方案，没有应用函数，因为矩阵对于慢速循环来说太大了。假设也有很多行，所以我想避免重复向量。

set.seed(1)

M = matrix(rpois(50,5),5,10)

v = c(3 ,   2 ,   7  ,  7 ,   4   , 4   , 7  ,  4  ,  5, 6)

M
     [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10]
[1,]    4    8    3    5    9    4    5    6    7     7
[2,]    4    9    3    6    3    1    5    7    6     1
[3,]    5    6    6   11    6    4    5    2    7     5
[4,]    8    6    4    4    3    8    3    6    5     6
[5,]    3    2    7    7    4    4    7    4    5     6

输出应该是

FALSE  FALSE FALSE FALSE TRUE

Answer 1

一种可能性是

rowSums(M == v[col(M)]) == ncol(M)
## [1] FALSE FALSE FALSE FALSE  TRUE

或者同样

rowSums(M == rep(v, each = nrow(M))) == ncol(M)
## [1] FALSE FALSE FALSE FALSE  TRUE

或者

colSums(t(M) == v) == ncol(M)
## [1] FALSE FALSE FALSE FALSE  TRUE

v[col(M)]只是 rep(v, each = nrow(M)) 的较短版本它创建了一个与 M 大小相同的向量(矩阵只是一个向量，尝试 c(M) )然后使用 == 将每个元素与其对应的元素进行比较.幸好 ==是一个泛型函数，它有一个 array方法(参见 methods("Ops") 和 is.array(M) )，它允许我们运行 rowSums (或 colSums )以确保我们的匹配数量为 ncol(M)