使用向量列表从矩阵列表中选择一列

Posted 2023-01-25

我想使用表示每个矩阵的列的布尔向量列表从不同维度的矩阵列表中选择列。

我尝试了不同的map和(s|l)apply组合，甚至经典的for循环，但我无法选择列。

使用此代码，您可以生成矩阵和布尔向量列表以进行实验：

matrices <- list(matrix(c(7,8,9,10), nrow=1), matrix(c(7,8,7,9,7,10,8,9,8,10,9,10), nrow=2), matrix(c(7,8,9,7,8,10,7,9,10,8,9,10), nrow=3))
listOfColumns <- list(c(FALSE,FALSE,FALSE,FALSE), c(FALSE,FALSE,FALSE,FALSE,FALSE,FALSE), c(TRUE,FALSE,FALSE,FALSE))

例如，使用上面的代码创建矩阵列表：

[[1]]
     [,1] [,2] [,3] [,4]
[1,]    7    8    9   10

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

[[3]]
     [,1] [,2] [,3] [,4]
[1,]    7    7    7    8
[2,]    8    8    9    9
[3,]    9   10   10   10

和布尔向量列表：

[[1]]
[1] FALSE FALSE FALSE FALSE

[[2]]
[1] FALSE FALSE FALSE FALSE FALSE FALSE

[[3]]
[1]  TRUE FALSE FALSE FALSE

结果应该是包含单个元素的列表：

[[1]]
[1] 7 8 9

答案

因为你的列表包含你根本不想保留的矩阵，我们可以先创建一个索引向量，它只选择listOfColumns的列表元素，其中至少有一个TRUE

(idx <- sapply(listOfColumns, any))
# [1] FALSE FALSE  TRUE

接下来，我们使用Map对剩余的矩阵进行子集化

Map(function(x, cols) x[, cols], x = matrices[idx], cols = listOfColumns[idx])
#[[1]]
#[1] 7 8 9

感谢@thelatemail提供的有用评论。