python range() 有重复项？

Question

每个人都知道可以使用 range 获取数字列表，如下所示；：

>>> list(range(5))
[0, 1, 2, 3, 4]

如果您想要每个数字的 3 个副本，则可以使用：

>>> list(range(5)) * 3
[0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 0, 1, 2, 3, 4]

但是有没有一种简单的方法使用 range< /code> 像这样重复复制？

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

示例：

sorted(list(range(5)) * 3)   # has unnecessary n * log(n) complexity
[x//3 for x in range(3*5)]   # O(n), but division seems unnecessarily complicated

Answer 1

您可以这样做：

>>> [i for i in range(5) for _ in range(3)]
[0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]

range(3) 部分应替换为您的重复次数...

顺便说一句，您应该使用生成器

---

只是为了使其更清楚， _ 是您不关心的变量名称（允许任何名称）。

此列表理解使用嵌套的 for 循环，就像这样：

for i in range(5):
    for j in range(3):
        #your code here

Answer 2

试试这个：

itertools.chain.from_iterable(itertools.repeat(x, 3) for x in range(5))

Answer 3

from itertools import chain, izip
list(chain(*izip(*[xrange(5)]*3)))

离开

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

列表，你就拥有了一台发电机。

编辑：甚至更好（省略对 izip 的函数调用）：

list(chain(*([x]*3 for x in xrange(5))))

Answer 4

在 numpy 的帮助下，有一种非常简单的方法可以做到这一点。示例：

>>> import numpy as np
>>> np.arange(5*3) // 3
array([0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4])

使用 range，您可以执行以下操作：

>>> list(map(lambda x: x // 3, range(5*3)))
[0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]

记住 // 执行严格的整数除法。

Answer 5

>>> from itertools import chain, izip, tee
>>> list(chain.from_iterable(izip(*tee(range(5), 3))))
[0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]

Answer 6

使用另一种方法的很酷的迭代器：

>>> from collections import Counter
>>> Counter(range(5) * 3).elements()

Answer 7

我喜欢保持简单:)

>>> sorted(list(range(5)) * 3)
[0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]

Answer 8

import itertools
[x for tupl in itertools.izip(*itertools.tee(range(0,5),3)) for x in tupl]

或者：

[x for tupl in zip(range(0,5), range(0,5), range(0,5)) for x in tupl]