如何在Python中重复每个元素的列表上进行迭代

2024-06-11 01:00:31 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用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

谢谢你的快速回答!


Tags: in元素列表forifdefnotiterable
3条回答

这个怎么样:

import itertools

def bicycle(iterable, repeat=1):
    for item in itertools.cycle(iterable):
        for _ in xrange(repeat):
            yield item

c = bicycle([1,2,3,4], 2)
print [c.next() for _ in xrange(10)]

编辑:合并bishanty's重复计数参数和Adam Rosenfield's list comprehension

itertools.chain.from_iterable(itertools.repeat(item, repeat) for item in itertools.cycle(l))

你可以很容易地用发电机:

def cycle(iterable):
    while True:
        for item in iterable:
            yield item
            yield item

x=[1,2,3]
c=cycle(x)

print [c.next() for i in range(10)]  // prints out [1,1,2,2,3,3,1,1,2,2]

相关问题 更多 >