Comment faire un objet itérable dans Python?

Dans Python, pour rendre un objet itérable, vous devez implémenter la méthode __iter__ et renvoyer un itérateur avec la méthode __Next__; Les étapes spécifiques sont les suivantes: 1. Définissez la méthode __iter__ dans la classe et renvoyez lui-même ou un autre objet Iterator qui implémente la méthode __Next__; 2. Si vous utilisez la fonction du générateur pour implémenter __iter__, vous pouvez gérer automatiquement la logique d'état et itérative via le mot clé de rendement; 3.

Dans Python, la clé de la création d'un objet itérable est de mettre en œuvre des méthodes de protocole spécifiques. Autrement dit, si un objet peut être utilisé dans for boucle ou transmis à iter() , il doit prendre en charge les protocoles itératifs.

Implémenter la méthode __iter__

Pour rendre un objet itérable, le moyen le plus direct est d'implémenter la méthode __iter__ pour cela et de retourner un itérateur. Cette méthode doit renvoyer un objet avec la méthode __next__ .

Par exemple:

 classe MyRange:
    def __init __ (self, start, fin):
        self.start = start
        self.end = fin

    def __iter __ (soi):
        Se retourner

    def __Next __ (self):
        Si self.start> = self.end:
            augmenter la pertinence
        courant = self.start
        self.start = 1
        Retour courant

De cette façon, vous pouvez utiliser votre classe comme vous le feriez avec range() :

 pour moi à MyRange (0, 3):
    Imprimer (i)
# Sortie: 0, 1, 2

Avis:

• Si votre classe agit comme un itérateur en même temps (comme dans l'exemple ci-dessus), alors son __iter__ revient.
• Sinon, vous pouvez écrire une classe Iterator distincte et renvoyer une instance de cette classe dans __iter__ .

Simplifier la logique avec le générateur

Si vous ne souhaitez pas gérer manuellement les exceptions de l'état et StopIteration , vous pouvez utiliser les fonctions du générateur pour simplifier l'implémentation de l'itérateur.

Par exemple, vous pouvez réécrire l'exemple ci-dessus comme celui-ci:

 classe MyRange:
    def __init __ (self, start, fin):
        self.start = start
        self.end = fin

    def __iter __ (soi):
        courant = self.start
        Tandis que le courant <self.end:
            courant de rendement
            courant = 1

Le mot-clé yield ici fera de __iter__ une fonction de générateur, renvoie automatiquement un itérateur, enregistrant les tracas de l'implémentation manuelle __next__ .

Assurer la compatibilité et le comportement correct

Après avoir mis en œuvre l'itérateur, vous devez faire attention à quelques choses faciles à ignorer:

• Chaque appel à iter(obj) doit renvoyer un nouvel itérateur afin que plusieurs boucles n'interfèrent pas entre elles.
• Si vous souhaitez que l'objet prenne en charge plusieurs traversées (par exemple, il peut être réutilisé en plusieurs for ), n'oubliez pas de réinitialiser l'état à chaque fois l'appel __iter__ .
• N'oubliez pas de lancer StopIteration pour mettre fin à l'itération, sinon il y aura des boucles infinies.

Fondamentalement, c'est tout. La maîtrise de ces deux points de base - mise en œuvre __iter__ et choisissant de simplifier avec un générateur - vous pouvez facilement transformer vos objets personnalisés en objets itérables.

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!