Python 生成器如何有效地展平任意巢狀清單？

使用Python 生成器展平不規則列表

展平任意嵌套的列表列表的問題很常見，但現有的解決方案經常因深度問題而失敗嵌套結構。特別是，像 [[[1, 2, 3], [4, 5]], 6] 這樣的清單無法使用大多數解決方案進行展平。

以下函數基於另一個問題的解決方案，可以有效地展平此類列表：

def flatten(x):
    result = []
    for el in x:
        if hasattr(el, "__iter__") and not isinstance(el, basestring):
            result.extend(flatten(el))
        else:
            result.append(el)
    return result

但是，雖然此函數有效，但可以使用生成器在可讀性和性能方面進行改進

帶有可迭代ABC 的Python 2

from collections import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, basestring):
            for item in flatten(x):
                yield item
        else:
            yield x

帶有可迭代ABC 和Yield From 的Python 3

from collections.abc import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, (str, bytes)):
            yield from flatten(x)
        else:
            yield x

這些基於生成器的解決方案提供了一種更簡潔、更有效的方法來展平不規則清單。它們的工作原理是迭代列表中的每個元素並遞歸地產生子列表的元素。任何不可迭代的元素（例如數字或字串）都會立即產生。

以上是Python 生成器如何有效地展平任意巢狀清單？的詳細內容。更多資訊請關注PHP中文網其他相關文章！