压扁不规则的列表列表

Question

压扁不规则的列表列表是的，我知道这个主题已经被覆盖过了（这里，这里，这里，这里），但据我所知，除了一个之外，所有解决方案都在这样的列表中失败：L = [[[1, 2, 3], [4, 5]], 6]期望的输出是什么[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 flatten(L)这是最好的型号吗？我忽略了什么吗？任何问题？

Answer 1

使用生成器函数可以使您的示例更容易阅读，并可能提高性能。

Python 2

def flatten(l):
    for el in l:
        if isinstance(el, collections.Iterable) and not isinstance(el, basestring):
            for sub in flatten(el):
                yield sub        else:
            yield el

我使用2.6中添加的Iterable ABC。

Python 3

在Python 3中，basestring不再是，但你可以使用元组str并bytes在那里获得相同的效果。

该yield from运营商从发生器一次一个返回的项目。这句法委派到子发生器在3.3加入

def flatten(l):
    for el in l:
        if isinstance(el, collections.Iterable) and not isinstance(el, (str, bytes)):
            yield from flatten(el)
        else:
            yield el