使用R中的字符向量提取矩阵中的特定元素

Question

我想用字符向量B中的信息从矩阵A中逐列提取特定元素(包含矩阵行名中的元素(，例如：

A <- matrix(seq(1,12),ncol=4)
rownames(A) <- letters[1:3]
A
[,1] [,2] [,3] [,4]
a    1    4    7   10
b    2    5    8   11
c    3    6    9   12
B <- c("a","c","c","b")

我想要1,6,9,11。谢谢：(

Answer 1

两种可能的方式：

> A[cbind(match(B, rownames(A)), seq_len(ncol(A)))]
[1]  1  6  9 11
> 
> diag(A[B, seq_along(B)]) # or diag(A[B, seq_len(ncol(A))])
[1]  1  6  9 11