如何在Python中重复地增加x个元素的个数y次

Question

我有x，这是所需元素的数量，y，这是所需元素的数量。你知道吗

例如：

>>> x = 5
>>> y = 2
>>> for i in range(y):
...  for j in range(x):
...   print j,
... 
0 1 2 3 4 0 1 2 3 4

预期输出为：

0 1 2 3 4
5 6 7 8 9

有人能帮我达到预期的产量吗？ for循环或列表理解任何方式都可以。 提前谢谢。你知道吗

Answer 1

您可以组合各种itertools迭代器来创建自己的生成器：

>>> from itertools import count, islice
>>> def foo(x, y):
...     elements = count()
...     for _ in range(y):
...         yield list(islice(elements, x))
...
>>> for es in foo(5, 2):
...     print(*es)
...
0 1 2 3 4
5 6 7 8 9