如何从嵌套列表中提取所有子列表?

时间:2019-11-03 00:54:05

标签: python python-3.x list

我有一个嵌套列表。例如:

['a', ['b', 'c', ['e', 'd']]]

我想获得一个列表,其中包含该列表和所有子列表分别作为元素。因此,预期结果是:

[['a', ['b', 'c', ['e', 'd']]], ['b', 'c', ['e', 'd']], ['e', 'd']]

我写了这个函数:

def extract(lst):
    result = []
    result.append(lst)
    for i in lst:
        if isinstance(i, list):
            result.append(i)
            extractt(i)
    return result

但是结果不是预期的。我该如何解决?

2 个答案:

答案 0 :(得分:3)

您可以将递归与生成器一起使用:

def get_lists(d):
  if isinstance(d, list):
     yield d
     for i in d:
        yield from get_lists(i)

print(list(get_lists(['a', ['b', 'c', ['e', 'd']]])))

输出:

[['a', ['b', 'c', ['e', 'd']]], ['b', 'c', ['e', 'd']], ['e', 'd']]

答案 1 :(得分:0)

我相信您的代码会丢弃对extract的递归调用的结果。您可以这样修改它:

def extract(lst):
    result = [lst]
    for i in lst:
        if isinstance(i, list):
            result += extract(i)
    return result