如何使用itertools重复每个Python列表的元素n次？

Question

我有一个数字列表: numbers = [1, 2, 3, 4].

我想有一个列表,他们重复这样的n时间(for n = 3):

[1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4].

问题是我只想用itertools它,因为我的性能受到很大限制.

我试着用这个表达式:

list(itertools.chain.from_iterable(itertools.repeat(numbers, 3)))

但它给了我这样的结果:

[1, 2, 3, 4, 1, 2, 3, 4, 1, 2, 3, 4]

这显然不是我需要的.

有没有办法itertools只使用,而不使用排序,循环和列表推导？我能得到的最接近的是:

list(itertools.chain.from_iterable([itertools.repeat(i, 3) for i in numbers])),

但它也使用列表理解,我想避免.

Answer 1

由于您不想使用列表推导,以下是一个纯(+ zip)itertools方法来执行此操作 -

from itertools import chain, repeat

list(chain.from_iterable(zip(*repeat(numbers, 3))))
# [1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]

Answer 2

首先，使用from中的函数itertools不一定比列表理解要快-您应该对这两种方法进行基准测试。（实际上，在我的机器上是相反的）。

纯列表理解方法：

>>> numbers = [1, 2, 3, 4]
>>> [y for x in numbers for y in (x,)*3]
[1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]

使用chain.from_iterable()带有发电机的表达：

>>> from itertools import chain, repeat
>>> list(chain.from_iterable(repeat(n, 3) for n in numbers))
[1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]