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

Question

我想從矩陣 A 中提取特定元素的列，其中包含來自字符向量 B 的信息（包含矩陣行名中的元素），例如：

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