具有位置向量的向量的子集列表

时间:2017-02-21 17:19:54

标签: r vector subset lapply

我有一个向量列表(mylist):

a <- c(1,2,3,4)
b <- c(5,6,7,8)
c <- c(9,10,11,12)
mylist <- list(a,b,c)

我还有一个位置向量(mypos):

mypos <- c(1,2,3)

我想使用mypos赋予元素的位置以对mypos的每个向量进行子集,以便它返回:

 [1] 1 6 11

我试过像这样使用lapply:

lapply(mylist, "[", mypos) 

但这会返回每个向量的元素1,2和3:

[[1]]
 [1] 1 2 3

[[2]]
 [1] 5 6 7

[[3]] 
 [1] 9 10 11

我也尝试过:

lapply(mylist, subset, mypos)

但是这会返回一个错误,该子集必须是逻辑

1 个答案:

答案 0 :(得分:5)

我们可以使用Map来提取&#39; mylist&#39;的相应元素。来自&#39; mypos&#39;

的索引
Map(`[`, mylist, mypos)

在OP的代码中,&#39; mypos&#39;在每个list元素中重复,导致从索引中提取所有元素。相反,它可以按顺序循环

lapply(seq_along(mylist), function(x) mylist[[x]][mypos[[x]]])