我有一个列表,(例如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
答案 0 :(得分:1)
我们需要遍历list
并将其子集
B1 <- lapply(A, function(x) x[idx==1,])
检查OP的结果是否相同
names(B) <- names(A)
identical(B, B1)
#[1] TRUE