python 自定义Iterator对象

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python 自定义Iterator对象，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含656字，纯文字阅读大概需要1分钟。

内容图文

from collections.abc import Iterator

class Company(object):
    def __init__(self, employee_list):
        self.employee = employee_list

    def __iter__(self):
        return MyIterator(self.employee) # 将list传入到这个里面自定义的Iterator里面，并且实现了__iter__，那么这个对象就可以是Iterator对象

class MyIterator(Iterator):
    def __init__(self, employee_list):
        self.iter_list = employee_list
        self.index = 0

    def __next__(self):
        #真正返回迭代值的逻辑
        try:
            word = self.iter_list[self.index]
        except IndexError:
            raise StopIteration # 这里必须抛出StopIteration，不然会一直跑
        self.index += 1
        return word

if __name__ == "__main__":
    company = Company(["tom", "bob", "jane"])

    for item in company:
        print (item)

内容总结

以上是互联网集市为您收集整理的python 自定义Iterator对象全部内容，希望文章能够帮你解决python 自定义Iterator对象所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端