使用列表理解时避免创建子列表

Question

如果我有这个列表理解：

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

我明白了

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

但我想得到：

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

注意到上面的例子只是一个最小的例子。

我已经阅读了很多 SO questions 解释如何展平列表列表的内容，但我没有找到关于如何避免在列表理解中创建子列表的内容。

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])))

使用列表理解时避免创建子列表

Avoid the creation of sublist when using a list comprehension

python

list-comprehension