Как перебрать список, повторяя каждый элемент в Python - PullRequest
8 голосов
/ 20 декабря 2008

Я использую 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

Спасибо за быстрые ответы!

Ответы [ 7 ]

13 голосов
/ 20 декабря 2008

Как насчет этого:

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)]

РЕДАКТИРОВАТЬ: включены параметр повторного подсчета Бишанты и понимание списка Адама Розенфилда .

6 голосов
/ 20 декабря 2008

Вы можете сделать это с помощью генератора довольно легко:

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]
2 голосов
/ 09 апреля 2016
itertools.chain.from_iterable(itertools.repeat(item, repeat) for item in itertools.cycle(l))
1 голос
/ 25 октября 2017

Я делаю это так:

from itertools import cycle, repeat, chain
flatten = chain.from_iterable # better name

def ncycle(xs, n):
    return flatten(repeat(x, n) for x in cycle(xs))

# example
for n,x in enumerate(ncycle('abcd', 2)):
    print(x, end=" ")
    if n > 9:
        print("")
        break
# output: a a b b c c d d a a b
1 голос
/ 22 декабря 2008
import itertools as it

def ncycle(iterable, n=1):
    if n == 1:
        return it.cycle(iterable)
    return it.cycle(it.chain(*it.izip(*([iterable]*n))))
1 голос
/ 20 декабря 2008

Решение должно быть что-то вроде

iterable = [1, 2, 3, 4]
n = 2

while (True):
  for elem in iterable:
    for dummy in range(n):
      print elem # or call function, or whatever

Редактировать: добавлено 'While (True)' для итерации до бесконечности.

0 голосов
/ 20 декабря 2008
[ "%d, %d" % (i, i) for i in [1, 2, 3, 4] * 4]

Последние 4 есть количество циклов.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...