zoukankan      html  css  js  c++  java
  • 你知道如何将python中嵌套的列表扁平化吗?

    楔子

    将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]
    
  • 相关阅读:
    VirtualBox 创建com对象失败
    大数据(十)
    HITCON 2014 已開始征求投稿计划书
    CSS
    工具
    工具
    Linux
    Python
    JavaScript
    JavaScript
  • 原文地址:https://www.cnblogs.com/traditional/p/12422934.html
Copyright © 2011-2022 走看看