你知道如何将python中嵌套的列表扁平化吗?

vs00ASPNET 2020-03-05

楔子

将python中嵌套的列表扁平化,指的就是:

lst = [1, [[[[3, 3], 5]]], [[[[[[[[[[[[[6]]]]], 8]]], "aaa"]]]], 250]]
# 类似于上面的列表,我们要得到下面的结果
[1, 3, 3, 5, 6, 8, "aaa", 250]

那么我们要如何做呢?

yield和yield from

关于如何把一个嵌套的列表扁平化,我们需要使用yield和yield from。这两位老铁的底层实现原理我不想深究,因为牵扯的东西太多了,况且针对目前的问题我们也用不到,这里我们简单的说一下。

def foo():
    yield [1, 2, 3]


def bar():
    yield from [1, 2, 3]


print(foo().__next__())  # [1, 2, 3]
print(bar().__next__())  # 1


"""
我们看到关于yield,它是把yield后面的整个元素都迭代出来
但是对于yield from,它的后面必须跟着一个可迭代的对象,然后依次迭代这个可迭代对象里面的每一个元素
"""

因此,我们就找到了方法。

def flat(l: list):
    for _ in l:
        if isinstance(_, list):
            yield from flat(_)
        else:
            yield _


lst = [1, [[[[3, 3], 5]]], [[[[[[[[[[[[[6]]]]], 8]]], "aaa"]]]], 250]]
print([_ for _ in flat(lst)])  # [1, 3, 3, 5, 6, 8, 'aaa', 250]

相关推荐