在Python中扁平化一个浅列表

是否有一种简单的方法可以用列表理解来扁平化一个可迭代对象列表，或者如果没有，你们都认为什么是扁平化这样一个浅列表的最好方法，平衡性能和可读性?

我尝试用一个嵌套的列表理解来扁平化这样一个列表，就像这样:

[image for image in menuitem for menuitem in list_of_menuitems]

但是我遇到了NameError的麻烦，因为名字‘menuitem’没有定义。在google和Stack Overflow上搜索之后，我用一个reduce语句得到了想要的结果:

reduce(list.__add__, map(lambda x: list(x), list_of_menuitems))

但是这个方法是相当不可读的，因为我需要调用list(x)因为x是Django QuerySet对象。

结论:

感谢每个为这个问题做出贡献的人。以下是我所学到的一份总结。我也把它变成了一个社区维基，以防其他人想要添加或纠正这些观察。

我原来的reduce语句是多余的，最好这样写:

>>> reduce(list.__add__, (list(mi) for mi in list_of_menuitems))

这是嵌套列表理解的正确语法(聪明的总结dF!)

>>> [image for mi in list_of_menuitems for image in mi]

但这两种方法都不如使用itertools.chain有效:

>>> from itertools import chain
>>> list(chain(*list_of_menuitems))

正如@cdleary所指出的那样，使用chain.from_iterable来避免*操作符魔法可能是更好的风格，就像这样:

>>> chain = itertools.chain.from_iterable([[1,2],[3],[5,89],[],[6]])
>>> print(list(chain))
>>> [1, 2, 3, 5, 89, 6]

当前回答

下面是一个使用集合处理多层列表的版本。Iterable:

import collections

def flatten(o, flatten_condition=lambda i: isinstance(i,
               collections.Iterable) and not isinstance(i, str)):
    result = []
    for i in o:
        if flatten_condition(i):
            result.extend(flatten(i, flatten_condition))
        else:
            result.append(i)
    return result

2014-12-15 06:03:05

其他回答

如果你要扁平化一个更复杂的列表与不可迭代元素或深度大于2，你可以使用以下函数:

def flat_list(list_to_flat):
    if not isinstance(list_to_flat, list):
        yield list_to_flat
    else:
        for item in list_to_flat:
            yield from flat_list(item)

它将返回一个生成器对象，您可以使用list()函数将其转换为列表。注意，yield from syntax可以从python3.3开始使用，但是可以使用显式迭代。例子:

>>> a = [1, [2, 3], [1, [2, 3, [1, [2, 3]]]]]
>>> print(list(flat_list(a)))
[1, 2, 3, 1, 2, 3, 1, 2, 3]

2015-04-12 01:05:48

这个解决方案适用于任意的嵌套深度——而不仅仅是一些(所有?)其他解决方案所限制的“列表的列表”深度:

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

它是递归，允许任意深度嵌套-直到你达到最大递归深度，当然…

2009-01-02 13:49:57

在我的脑海中，你可以消去lambda

reduce(list.__add__, map(list, [mi.image_set.all() for mi in list_of_menuitems]))

或者甚至删除地图，因为你已经有了一个列表:

reduce(list.__add__, [list(mi.image_set.all()) for mi in list_of_menuitems])

你也可以将它表示为列表的和:

sum([list(mi.image_set.all()) for mi in list_of_menuitems], [])

2009-01-02 05:55:01

在Python 2或3中实现这一点的最简单方法是使用pip install morph使用morph库。

代码是:

import morph

list = [[1,2],[3],[5,89],[],[6]]
flattened_list = morph.flatten(list)  # returns [1, 2, 3, 5, 89, 6]

2015-12-15 20:00:04

是什么:

from operator import add
reduce(add, map(lambda x: list(x.image_set.all()), [mi for mi in list_of_menuitems]))

但是，Guido建议不要在一行代码中执行太多操作，因为这会降低可读性。在单行中执行与在多行中执行您想要的操作相比，性能收益最小(如果有的话)。

2009-01-02 06:28:58

在Python中扁平化一个浅列表

推荐文章

最新文章

标签