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

Question

我在列表中有一个列表。

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

For example: L=[['READ',[A,B],'2'],['WRITE',[C,D],'2']]

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

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

Answer 1

使用常见的展平功能：

import collections
def flatten(l):
    for el in l:
        if isinstance(el, collections.Iterable) and not isinstance(el, basestring):
            for sub in flatten(el):
                yield sub
        else:
            yield el

您可以展平列表，然后检查 READ 是否在其中。

>>> 'READ' in flatten(L)
True

Answer 2

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

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))