通过重复Python中的单个元素来扩展列表

Question

有没有更简单的方法来实现以下目标？

lst = []
repetitions = 3
for elem in range(3):
    lst  += [elem] * repetitions

这会将 [0,1,2] 变为 [0,0,0,1,1,1,2,2,2]

Answer 1

您可以使用带有两个循环的列表理解：

>>> [elem for elem in range(3) for _ in range(repetitions)]
[0, 0, 0, 1, 1, 1, 2, 2, 2]