python - 在列表理解中使用列表解包来展平嵌套列表

Question

这个问题已经有答案了:

How do I make a flat list out of a list of lists? (36 个回答)

已关闭 5 年前。

我看到了this合并列表的解决方案，例如 a = [1,2,3], b = [4,5,6]使用res = [*a, *b] .

假设我有一个子列表列表，例如 ls = [a,b]是否可以做类似 res = [*i for i in ls] 的事情？

该特定行自 SyntaxError: iterable unpacking cannot be used in comprehension 起无效。可以做类似的事情吗？

如果没有，如何轻松创建一个包含子列表中所有元素的列表？

使用 python 3.5.3

Answer 1

不，我不相信他们已经在理解中添加了对列表解包的支持。

作为替代方案，您可以使用itertools.chain:

>>> from itertools import chain
>>> list(chain.from_iterable([a, b]))
[1, 2, 3, 4, 5, 6]

或者，嵌套循环列表理解:

>>> [y for x in [a, b] for y in x]
[1, 2, 3, 4, 5, 6]