for循环的底层原理

80 阅读 0 评论 53 点赞

我是靠谱客的博主温柔花卷，最近开发中收集的这篇文章主要介绍for循环的底层原理，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

# 自定义可迭代对象：在类里面提供了__iter__方法创建的对象就是可迭代对象
# 自定义迭代器对象：在类里面提供__iter__和__next__的方法创建的对象就是迭代器对象
# 迭代器的作用：记录当前数据的位置以后获取下一个位置的值
# iter函数:获取可迭代对象的迭代器，会调用可迭代对象身上的__iter__方法
# next函数:获取迭代器中下一个值，会调用迭代器对象身上的__next__方法
# for 循环的本质: 如果遍历的对象是可迭代对象那么会通过iter函数获取迭代器，然后通过next函数获取迭代器中下一个值
# 如果遍历的对象是迭代器，那么会通过next函数获取迭代器中下一个值
# 提醒: for循环内部自己捕获停止迭代的异常，使用while True 取值自己需要捕获停止迭代异常

# 自定义可迭代对象
class Student_List(object):

# 提供构造方法，创建存储数据的容器->列表
def __init__(self):
self.my_list = []

# 向列表中添加学生信息
def add_student(self,student):
self.my_list.append(student)

# 提供可以迭代的方法
def __iter__(self):
student_iterator = Student_Iterator(self.my_list)
return student_iterator

# 自定义迭代器
class Student_Iterator(object):
# 把自定义的可迭代对象传过来
def __init__(self, my_list):
self.my_list = my_list
# 设置默认下标为0
self.current_index = 0

# 提供可以迭代的方法,返回自身即可
def __iter__(self):
return self

# 采用__next__()方法获取对象中下一个值
def __next__(self):
# 如果下标小于列表长度，则返回取到相应下标列表的值
if self.current_index < len(self.my_list):
# 要在return之前将下标值加1，因为若放在return之后就不执行了
self.current_index += 1
# 因为先将下标进行了加1操作，所有这里要减1
return self.my_list[self.current_index-1]
else:
print('下标越界')
raise StopIteration

if __name__ == '__main__':
student_list = Student_List()
student_list.add_student('ss')
student_list.add_student('xx')

# 遍历
for student in student_list:
print(student)

转载： https://blog.csdn.net/longting_/article/details/80465642