python - 使用列表推导时避免创建子列表

Question

如果我有这个列表理解：

[list(range(0,x)) for x in [1, 2, 3]] 

我明白了

[[0], [0, 1], [0, 1, 2]]

但我想得到：

[0, 0, 1, 0, 1, 2]

请注意，上面的示例只是一个最小示例。

我已经阅读了很多解释如何展平列表列表的SO 问题，但我没有发现如何避免在列表理解中创建子列表。

标签： pythonlist-comprehension

Answer 1

使用嵌套理解 à la：

[y for x in [1,2,3] for y in range(x)]
# [0, 0, 1, 0, 1, 2]

或者，如果您喜欢实用程序和神秘简洁：

from itertools import chain

[*chain(*map(range, [1, 2, 3]))]
# or, as the traditionalists would suggest:
# list(chain.from_iterable(map(range, [1, 2, 3])))