r - 给定另一个列表中的位置，从列表中提取元素

Question

我有一个矩阵列表如下。

listMat <- list(A=matrix(1:9, ncol=3), B=matrix(10:18, ncol=3))
listMat 

$A
     [,1] [,2] [,3]
[1,]    1    4    7
[2,]    2    5    8
[3,]    3    6    9

$B
     [,1] [,2] [,3]
[1,]   10   13   16
[2,]   11   14   17
[3,]   12   15   18

我想提取矩阵 A 的第 1 列和第 2 列以及 B 的第 3 列。这些位置在列表中给出。

listPosition <- list(A=c(1,2), B=c(1))
listPosition 
$A
[1] 1 2

$B
[1] 1

有没有办法提取元素而不是使用以下命令

  listMat[["A"]][,1:2] 
  listMat[["B"]][,1] 

Answer 1

我们可以使用Map利用listPosition中的索引提取list中对应的matrix元素

Map(function(x, y) x[, y, drop = FALSE], listMat, listPosition)