Exemple de méthode pour implémenter un itérateur en python

Présentation

Les itérateurs sont un moyen d'accéder aux éléments d'une collection. L'objet itérateur commence à accéder à partir du premier élément de la collection jusqu'à ce que tous les éléments aient été accédés. Les itérateurs ne peuvent qu'avancer et non reculer.

Évaluation paresseuse ou évaluation paresseuse (Évaluation paresseuse)

Les itérateurs ne vous obligent pas à préparer à l'avance tous les éléments de l'ensemble du processus d'itération. Un élément n'est évalué que lors d'une itération vers lui, et avant ou après cela, l'élément peut ne pas exister ou être détruit. Cette fonctionnalité le rend particulièrement adapté pour parcourir des collections énormes ou infinies.

Aujourd'hui, j'ai créé une classe d'entité, à peu près comme suit :

class Account():
 def __init__(self,
   account_name,
   account_type,
   account_cost,
   return_amount=0):
 self.account_name = account_name # 账户名
 self.account_type = account_type # 账户类型
 self.account_cost = account_cost # 月结费用
 self.return_amount = return_amount # 返还金额

Ensuite, créez une Liste d'entités :

accounts = [Account("张三", "年费用户", 450.00, 50),
  Account("李四", "月结用户", 100.00),
  Account("杨不悔", "月结用户", 190.00, 25),
  Account("任我行", "月结用户", 70.00, 10),
  Account("凌未风", "年费用户", 400.00, 40)]

Je souhaite exécuter la fonction next(), c'est-à-dire "suivant" lorsque cela est nécessaire pour obtenir l'élément suivant dans la Liste un élément.

Testez-le directement :

Il s'avère que List ne prend pas en charge la fonctionnalité next(). Pour le moment, List n’est qu’un itérable, pas un itérateur.

La différence entre iterable et iterator est la suivante :

1. iterable - implémente uniquement les objets __iter__ ;

   iterator - un objet qui implémente les méthodes __iter__ et __next__.

2. Parmi eux,
renvoie l'objet itérateur et

__iter____next__1. Faire de la liste un itérateur

Pour faire de la liste des comptes précédents un itérateur, juste une simple fonction
:

iter()

accounts_iterator = iter(accounts)
(next(accounts_iterator)).account_name

Une fonction si simple, on estime qu'il existe encore de nombreux développeurs Python ne savent pas ?

2. Objet itérateur personnalisé

En terme d'expansion, comment définir votre propre objet itérateur ? En fait, cela signifie implémenter les méthodes
et

__iter__Nous définissons ensuite une classe AccountIterator : __next__

class AccountIterator():
 def __init__(self, accounts):
 self.accounts = accounts # 账户集合
 self.index = 0

 def __iter__(self):
 return self

 def __next__(self):
 if self.index >= len(self.accounts):
  raise StopIteration("到头了...")
 else:
  self.index += 1
  return self.accounts[self.index-1]

Le résultat courant est le suivant :

Après un moment de lancer, la fonction next() a été réalisée. Python possède de nombreuses fonctions inattendues qui attendent que nous continuions à les explorer. C'est peut-être le charme et le côté geek de Python.