时间戳

首页 > 解决方案 > R：循环列表列表以检索包含命中的子列表的标题

r - R：循环列表列表以检索包含命中的子列表的标题

问题描述

我在 R 中有一个列表列表。列表列表中的每个子列表都包含多个元素。这些子列表不一定都具有相同的长度。所有子列表都有一个特定的标题名称。像这样：

#create list of lists
vector1 = c("apple","banana","cherry")
vector2 = c("banana","date","fig")
vector3 = c("fig","jackfruit","mango","plum")
listoflists  = list(vector1 , vector2, vector3)
names(listoflists) = c("listA", "listB", "listC")

列表列表如下所示：

listoflists

$listA
[1] "apple"  "banana" "cherry"

$listB
[1] "banana" "date"   "fig"   

$listC
[1] "fig"       "jackfruit" "mango"     "plum"

接下来，我有一个向量，其中包含也可以在子列表中找到的元素。像这样：

wanted = c("apple","banana","fig")
wanted
[1] "apple"  "banana" "fig"

对于想要的向量中的每个元素，我想提取包含此特定元素的列表列表中每个子列表的标题名称。对于此处提供的示例，输出应如下所示：

#desired output
apple  listA
banana listA listB
fig    listB listC

我想过把它放到一个 for 循环中来获得这样的东西：

output_list = list()
for (i in wanted){
  output = EXTRACT LIST HEADER WHEN i IS PRESENT IN SUBLIST
  output_list[[i]] = output
}

但是，尚不清楚我是否可以，如果可以，如何循环遍历列表列表以仅提取那些包含 vector Want 中元素的子列表的标题名称。我研究了使用该unlist功能，但这似乎对这个问题没有用。我查看了 stackoverflow 以及其他论坛，但找不到任何概述类似问题的问题。因此，如果有人能指出我解决这个问题的正确方向，那将非常有帮助。

已经谢谢了！

标签： rlistloopsvectorsublist

解决方案

这是另一个基本 R 选项

u <- unlist(listoflists)
sapply(wanted, function(x) rep(names(listoflists),lengths(listoflists))[u %in% x])

这使

$apple
[1] "listA"

$banana
[1] "listA" "listB"

$fig
[1] "listB" "listC"

推荐阅读