Revenir en arrière dans l'itérateur python

Vous pouvez boucler pour toujours en utilisant itertools.cycle() et vous pouvez utilisez itertools.chain() pour ajouter la condition de départ comme:

import itertools as it
for i in it.chain((0,), it.cycle(range(1, 6))):
    print(i)

Résultats:

0
1
2
3
4
5
1
2
3
4
5

Vous pouvez reproduire une grande partie du style C pour la boucle à l'aide d'une classe itérateur.

Par exemple:

class cFor:   
    def __init__(self,start,condition,step=1):
        self.index     = start
        self.nextIndex = start
        self.doLoop    = condition
        if type(condition) == type(start):
            self.doLoop  = lambda x:x < condition
        self.getNext   = step
        if type(step) == type(start):
            self.getNext = lambda x:x + step

    def __iter__(self):
        return self

    def next(self) : return self.__next__()

    def __next__(self):
        if self.doLoop(self.nextIndex):
            self.index     = self.nextIndex
            self.nextIndex = self.getNext(self.index)
            return self.index
        raise StopIteration()

Cela vous permettra d'effectuer des boucles simples:

for i in cFor(0,10,2) : print(i)
# 0, 2, 4, 6, 8

Et, à l'aide de lambdas, vous pouvez fournir votre propre calcul de condition et d'incrémentation:

for i in cFor(1, lambda i:i<=10, lambda i:i+3): print(i)
# 1, 4, 7, 10

Ou appliquez des calculs "en ligne" plus complexes:

[ a for a,b in cFor((1,1),lambda ab:ab[1]<30,lambda ab:(ab[1],sum(ab))) ]
# [ 1, 1, 2, 3, 5, 8, 13 ]

Pour pouvoir modifier l'index (ou la condition ou l'incrément) dans la boucle, vous pouvez créer une fonction de générateur qui renvoie à la fois l'index en cours et l'instance de l'itérateur. Cela vous permettra de prendre le contrôle des propriétés de l'itérateur dans le corps de la boucle.

def cForLoop(start,condition,step=1):
    loop = cFor(start,condition,step)
    while True : yield loop.next(), loop

for i,loop in cForLoop(0,10):
    print(i)
    if i == 5 : loop.nextIndex = 1
# 0, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ....

Vous pouvez, par exemple, l'utiliser pour implémenter un simple tri à bulles:

arr = [3,7,4,2,8,1,6]

for i,loop in cForLoop(1,len(arr)):
    if arr[i-1] > arr[i] :
        arr[i],arr[i-1] = arr[i-1],arr[i]
        loop.nextIndex = max(1,i-1) 

# arr == [1, 2, 3, 4, 6, 7, 8]

@Stephen J'essaye sans chaîne et je travaille bien. merci gentil code

import itertools as it
for i in it.cycle(range(1, 6)):
    print(i)