学习笔记(06):Python实战编程-自定义迭代

立即学习:https://edu.csdn.net/course/play/19711/255553?utm_source=blogtoedu

python类中，可以__iter__()、__next__(),进行迭代，使用for像数组、字典一样循环。

例子：

# -*- coding=utf-8 -*-
import os, sys


class Message:
    def __init__(self, max):
        self.__max = max
        self.__index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.__index >= self.__max:
            return -1
        else:
            val = self.__max - self.__index
            self.__index += 1
            return val


if __name__ == "__main__":
    print("进入主函数......")

    msg = Message(10)
    for v in msg:
        if v == -1:
            break
        print(v, end="、")

执行结果:

进入主函数......
10、9、8、7、6、5、4、3、2、1、