矩阵和输出索引的子集矩阵

时间:2016-04-20 19:35:23

标签: r subset

我有一个矩阵,我想要分组。当我对它进行子集时,我想要的输出是其rowname的值。

m
  x y
A 2 2
B 2 2
C 1 3
D 1 1
E 1 2

m2
  x y
1 1 2

这是我目前所拥有的,显然不是正确的输出。这看起来很简单,但我还没弄明白。

which(m2==m, arr.ind=TRUE)
  row col
1   1   2

输出应为

output
"E"

m2可以有多于1行到子集m。

1 个答案:

答案 0 :(得分:1)

如果你将m2的每一列都设为一个向量,你可以借用in this answer所示的技巧。请注意,我在m2添加了一行以更好地说明问题:

m <- matrix(c(2,2,2,2,1,3,1,1,1,2), 
            nrow = 5, ncol = 2, byrow = T, 
            dimnames = list(LETTERS[1:5], c("x", "y")))

m2 <- matrix(c(1,2,2,2), nrow = 2, ncol = 2, byrow = T, 
             dimnames = list(c(1,2), c("row", "col")))

apply(m2, 1, function(i) rownames(m)[colSums(t(m) == as.vector(i)) == ncol(m)])

# $`1`
# [1] "E"
# 
# $`2`
# [1] "A" "B"

建议De Morgan's Law的较短in this answer版本也会应用 ......

apply(m2, 1, function(i) rownames(m)[!colSums(t(m) != as.vector(i))])