压扁不规则的列表列表

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 2def 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