我使用Python无限地对列表进行迭代,重复列表中的每个元素多次。例如,给定列表:
l = [1, 2, 3, 4]
我想输出每个元素两次,然后重复循环:
1, 1, 2, 2, 3, 3, 4, 4, 1, 1, 2, 2 ...
我知道从哪里开始:
def cycle(iterable):
if not hasattr(cycle, 'state'):
cycle.state = itertools.cycle(iterable)
return cycle.next()
>>> l = [1, 2, 3, 4]
>>> cycle(l)
1
>>> cycle(l)
2
>>> cycle(l)
3
>>> cycle(l)
4
>>> cycle(l)
1
但我该如何重复每一个元素呢?
编辑
为了澄清这一点,应该无限循环。我还使用了重复元素两次作为最短的示例- 我真的想重复每个元素n次。
更新
你的解决方案能让我找到我想要的吗:
>>> import itertools
>>> def ncycle(iterable, n):
... for item in itertools.cycle(iterable):
... for i in range(n):
... yield item
>>> a = ncycle([1,2], 2)
>>> a.next()
1
>>> a.next()
1
>>> a.next()
2
>>> a.next()
2
>>> a.next()
1
>>> a.next()
1
>>> a.next()
2
>>> a.next()
2
谢谢你的快速回答!
这个怎么样:
编辑:合并bishanty's重复计数参数和Adam Rosenfield's list comprehension。
你可以很容易地用发电机:
相关问题 更多 >
编程相关推荐