Comment fonctionnent les itérateurs Python : un guide des méthodes `__iter__` et `__next__` ?

Comment fonctionnent les itérateurs Python : un guide complet

Les itérateurs sont des éléments essentiels de la programmation Python, permettant un parcours efficace et concis de séquences ou la génération de valeurs à la volée. Cet article explore comment créer un itérateur de base en Python, en se concentrant sur les concepts fondamentaux et la mise en œuvre.

Comprendre le protocole d'itérateur

Les itérateurs Python suivent un protocole spécifique, défini par les méthodes __iter__() et __next__(). La méthode __iter__() renvoie l'objet itérateur, qui sert de point d'entrée pour l'itération. À l'inverse, la méthode __next__() renvoie la valeur suivante dans la séquence ou déclenche une exception StopIteration lorsqu'il n'y a plus de valeurs.

Création d'un itérateur personnalisé

Considérez la Exemple de classe suivante qui "contient" logiquement des valeurs :

class Example:
    def __init__(self, values):
        self.values = values

Pour implémenter un itérateur pour cela classe, nous définissons les méthodes __iter__() et __next__() :

class Example:
    def __init__(self, values):
        self.values = values

    def __iter__(self):
        return self

    def __next__(self):
        if self.values:
            return self.values.pop(0)
        else:
            raise StopIteration

Dans cet exemple, l'itérateur renvoie les valeurs de la liste de valeurs en les affichant une par une. Lorsque la liste est vide, elle lève une exception StopIteration.

Utilisation de l'itérateur

Maintenant, nous pouvons utiliser l'itérateur comme suit :

e = Example([1, 2, 3])

for value in e:
    print("The example object contains", value)

Ce code parcourra les valeurs [1, 2, 3] et imprimera chaque valeur dans le console.

Itérateurs plus complexes

Les itérateurs ne se limitent pas à accéder à des attributs ou à des valeurs spécifiques d'une instance. Ils peuvent également contrôler la provenance des valeurs ou même les calculer à la volée. Par exemple, la classe Counter suivante génère une plage de valeurs :

class Counter:
    def __init__(self, low, high):
        self.current = low - 1
        self.high = high

    def __iter__(self):
        return self

    def __next__(self): 
        self.current += 1
        if self.current < self.high:
            return self.current
        raise StopIteration

En appelant la méthode __iter__() sur cette classe, nous obtenons un itérateur qui générera des valeurs dans la plage spécifiée.

Conclusion

Les itérateurs Python fournissent un mécanisme puissant et flexible pour itérer sur des séquences ou générer des valeurs. Comprendre le protocole de l'itérateur et comment implémenter des itérateurs personnalisés permet aux développeurs de créer un code efficace et polyvalent.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!