编写列表理解以展平嵌套列表[重复]

Question

我有一个包含列表和数字的嵌套列表。

nested = [[1, 2, 3], [4, 5, 6], [7, 8], [9], 10, 11]

嵌套列表只包含数字，它们永远不会包含更多列表。

是否可以编写一个列表推导来从“嵌套”列表中创建新列表，产生以下输出？

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]

这是我的尝试（代码不工作）

[num if isinstance(item, list) for num in item else item for item in nested]

Answer 1

您需要将您的实例测试放在一个循环中，这样您就可以从嵌套列表中提取元素：

[num for item in nested for num in (item if isinstance(item, list) else (item,))]

演示：

>>> nested = nested = [[1, 2, 3], [4, 5, 6], [7, 8], [9], 10, 11]
>>> [num for item in nested for num in (item if isinstance(item, list) else (item,))]
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]

如果您首先将其表达为一组for 循环，将会有所帮助；您的尝试基本上是这样做的：

for item in nested:
    _element = num if isinstance(item, list) for num in item else item
    result.append(_element)

这不是真正有效的 Python。

我上面的列表理解是这样做的：

for item in nested:
    _iterable = item if isinstance(item, list) else (item,)
    for num in _iterable:
        _element = num
        result.append(_element)

Answer 2

如果您的列表有任意嵌套，最好使用递归函数 (referring to one of my other answers)：

>>> def flat_gen(x):
...     def iselement(e):
...         return not(isinstance(e, collections.Iterable) and not isinstance(e, str))
...     for el in x:
...         if iselement(el):
...             yield el
...         else:
...             yield from flat_gen(el)   
... 
>>> nested = [[1,2,3] , [4,5,6] , [7,8] , [9,] , 10 , 11 ]

>>> import collections
>>> list(flat_gen(nested))
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]

如果您只是在简单列表或元组（任何可迭代容器）和非列表或元组对象（即不可迭代对象和没有更深嵌套的可迭代容器）之间混合嵌套，您可以使用带有 @987654324 的生成器表达式@：

>>> import itertools as it
>>> list(it.chain.from_iterable(item if isinstance(item,collections.Iterable) and not isinstance(item, str) else [item] for item in nested))
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]