手写一个python迭代器过程详解

Python  /  管理员 发布于 7年前   201

分析

• 我们都知道一个可迭代对象可以通过iter()可以返回一个迭代器。
• 如果想要一个对象称为可迭代对象，即可以使用for，那么必须实现__iter __()方法。
• 在一个类的实例对象想要变成迭代器，就必须实现__iter__()和__next__()方法。
• 调用iter()时，在对象内部默认调用__iter__()，即__iter__()的返回值应该是一个迭代器。
• for的每次循环中或者next()时，都是自动调用迭代器的__next__()方法，并有一个返回值。

实现

class Classmate: def __init__(self):  self.names = []  self.num = 0 def add(self, name):  self.names.append(name) def __iter__(self):  return self def __next__(self):  if self.num < len(self.names):   ret = self.names[self.num]   self.num += 1   return ret  else:   raise StopIterationc = Classmate()c.add("张三")c.add("李四")c.add("王五")for i in c: print(i)"""张三李四王五"""

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。