如何在不丢失元素名称的情况下获取列表中元素的子集?

时间:2016-06-09 11:38:18

标签: r list lapply

我有一个列表,(例如A),其中包含多个元素。每个元素具有相同的尺寸(例如100×200)。我有一个矢量,idx,长度为100,行数与列表中的元素相同。我想创建一个新列表,它是列表A的一个子集,例如idx == 1。我使用了lapply,它似乎工作但我丢失了列表中的元素名称。你能建议一个更好的方法来实现这个目标吗?非常感谢

 # Generate a list A
 A = list()
 for (i in 1:26){
   A[[LETTERS[i]]] = matrix(runif(100*200,0,1),100,200)
 }
 # Generate a vector with 100 elements with integers between 1 and 5
 idx = sample(1:5, 100, replace=T)
 # Create subset list B from list A, where rows are idx==1
 B = lapply(1:26, function(x) A[[x]][idx==1,])
 # However, in list B, I have lost all the element names

1 个答案:

答案 0 :(得分:1)

我们需要遍历list并将其子集

B1 <- lapply(A, function(x) x[idx==1,])

检查OP的结果是否相同

names(B) <- names(A)
identical(B, B1)
#[1] TRUE