r - 比较两个矩阵，保持一个矩阵中的值在另一个矩阵中为 TRUE

Question

这似乎是一项简单的任务，在查找此处和其他地方后，我没有在 R 上找到解决方案。我有两个矩阵，一个包含字符串值，另一个包含逻辑值。

a <- matrix(c(
              "A", "B", "C"
              ))

b <- matrix(c(
              T, F, T
              ))

> b
      [,1]
[1,]  TRUE
[2,] FALSE
[3,]  TRUE
> a
     [,1]
[1,] "A"
[2,] "B"
[3,] "C"

我需要创建第三个矩阵，将第一个矩阵中的值保留在第二个矩阵中为 TRUE，并在其余矩阵中保留 NA，如下所示:

> C
      [,1]
[1,]  "A"
[2,]  NA
[3,]  "C"

• 如何实现上述结果？

Answer 1

C <- matrix(a[ifelse(b, T, NA)], ncol = ncol(a))