如何在嵌套列表中搜索项目？

Question

我需要在嵌套列表中搜索特定列表。

具体地说，我需要检查嵌套在主列表中的一个列表中的第一项是否为=到“ A”，并返回该特定列表。

我不知道从哪里开始

list1 = ['A', 'B', 'C']
list2 = ['D', 'E', 'F']
list3 = [list1, list2]

我要搜索字母A并返回list1

Answer 1

使用递归函数：

def checkInNestedList(val, lst):
    if val in lst:
            return lst
    for i in lst:
            if type(i) == list:
                    return checkInNestedList(val, i)
    return None

---

>>> l = [['A', 'B', 'C'], ['D', 'E', 'F']]
>>> checkInNestedList('A', l)
['A', 'B', 'C']
>>> l = [[[[['A', 'B'], ['C']], ['D', 'E', 'F']]]]
>>> checkInNestedList('A', l)
['A', 'B']
>>>