找到等于vector的所有矩阵行

Question

假设我有以下矩阵:

cm<-structure(c(100, 200, 400, 800, 100, 200, 400, 800, 100, 200, 
400, 800, 100, 200, 400, 800, 100, 200, 400, 800, 0, 0, 0, 0, 
0.5, 0.5, 0.5, 0.5, 1, 1, 1, 1, 0, 0, 0, 0, 0.5, 0.5, 0.5, 0.5, 
-0.4, -0.4, -0.4, -0.4, -0.4, -0.4, -0.4, -0.4, -0.4, -0.4, -0.4, 
-0.4, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 
1, 1, 1, 1, 1, 1, 1, 1, 1), .Dim = c(20L, 4L), .Dimnames = list(
    NULL, c("Var1", "Var2", "Var3", "n1")))

以及从中衍生的另一个矩阵:

a4<-data.matrix(unique(cm[,1:3]))

现在,我想找到cm前三列相等的所有行a4[1,],但是做了直观的事情:

a5<-which(cm[,1:3]==a4[1,])

失败(R 3.1.3).例如a5[2]是13,但第13行cm[,1:3]是不一样的a4[1,].

Answer 1

该功能row.match在prodlim简单易用,非常适合您的问题.

library(prodlim)
row.match(a4[1,], cm[,1:3])
[1] 1

Answer 2

使用apply和all.equal比较每一行与目标行.使用的问题==是它只检查它回收矢量的元素进行比较,而你想看看行矢量中的所有值是否匹配,a4[1,]所以你应该使用all.equal.结果是它的返回值不是逻辑,而是描述对象之间差异的字符串,这使得它比==单独使用更麻烦:

which(apply(cm, 1, function(x) all.equal(x[1:3], a4[1,])) == "TRUE")
# [1] 1

您还可以使用以下代码identical来简化all.equal:

which(apply(cm, 1, function(x) identical(x[1:3], a4[1,])))
# [1] 1

然后提取:

cm[apply(cm, 1, function(x) identical(x[1:3], a4[1,])),,drop=FALSE]
#      Var1 Var2 Var3 n1
# [1,]  100    0 -0.4  1

为了弄清楚究竟发生了什么,请考虑==传递矩阵参数时隐含的内容:

which(cm[,1:3]==a4[1,])
# [1]  1 13 23 35 42 45 48 51 53 56 59

该结果与将矩阵转换为向量相同:

as.vector(cm[,1:3])
#  [1] 100.0 200.0 400.0 800.0 100.0 200.0 400.0 800.0 100.0 200.0 400.0 800.0 100.0 200.0 400.0 800.0 100.0 200.0 400.0 800.0   0.0   0.0   0.0   0.0   0.5   0.5   0.5
# [28]   0.5   1.0   1.0   1.0   1.0   0.0   0.0   0.0   0.0   0.5   0.5   0.5   0.5  -0.4  -0.4  -0.4  -0.4  -0.4  -0.4  -0.4  -0.4  -0.4  -0.4  -0.4  -0.4   0.0   0.0
# [55]   0.0   0.0   0.0   0.0   0.0   0.0
which(as.vector(cm[,1:3])==a4[1,])
# [1]  1 13 23 35 42 45 48 51 53 56 59

因此,位置是矢量表示内的位置cm,而不是矩阵表示中的行.==当尝试比较不等长的向量或者一个向量的长度不是另一个向量的倍数时,比较也可能是危险的(再次对上面提到的回收做),这将产生警告:

1:2 == 1:3
# [1]  TRUE  TRUE FALSE
# Warning message:
# In 1:2 == 1:3 :
#   longer object length is not a multiple of shorter object length

使用回收时没有警告:

1:2 == 1:6
# [1]  TRUE  TRUE FALSE FALSE FALSE FALSE