分析

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

    实现

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    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 StopIteration

    c = Classmate()
    c.add("张三")
    c.add("李四")
    c.add("王五")
    for i in c:
    print(i)
    """
    张三
    李四
    王五
    """

__END__

三国小梦
文章作者:三国小梦
文章出处手写一个python迭代器
作者签名:简单地活着, 肆意又精彩.
关于主题Hexo - Live For Code
版权声明:文章除特别声明外,均采用 BY-NC-SA 许可协议,转载请注明出处