python - 在列表列表中查找元素

Question

我有一个列表中的列表。

我想查找它是否包含感兴趣的项目。

例如:L=[['READ',[A,B],'2'],['WRITE',[C,D],'2']]

现在，我有一个字符串 str=READ，我想遍历两个列表，包括子列表以查找是否存在这样的元素。有没有办法不使用索引就可以做到这一点？

我不想使用索引，因为无法保证列表长度保持不变。

Answer 1

如果正在搜索的数据结构中没有循环，这是一个简单的递归问题:

def find(x, L):
    return x in L or any(find(x, sublist)
                         for sublist in L
                         if isinstance(sublist, list))

如果相反，数据结构中可能存在循环，那么您必须防止进入无限递归

def find(x, L, seen=None):
    if seen is None:
        seen = set()
    if id(L) in seen:
        # Avoid infinite recursion
        return False
    seen.add(id(L))
    return x in L or any(find(x, sublist, seen)
                         for sublist in L
                         if isinstance(sublist, list))