我们在 Python 中使用 for 循环迭代时到底做了什么
转载请注明出处:https://blog.csdn.net/jpch89/article/details/85532979
-
使用
for
循环迭代一般写成:for i in x: ...
-
首先看
x
是否实现了__iter__
方法,这个方法的返回值是一个迭代器。- 如果
x
实现了x.__iter__()
方法,那么使用iter(x)
得到x
的迭代器it_x
。 - 然后对迭代器
it_x
不断地调用next(it_x)
,把每次返回的元素给i
。 - 最终迭代器元素耗尽,捕获到
StopIteration
异常,终止迭代。
- 如果
-
如果
x
没有实现__iter__
方法,那么看它是否实现了__getitem__
方法。
(序列类型Sequence
都实现了__len__
和__getitem__
方法,所以序列类型的对象都是可以用于for
循环的。)- 如果实现了
__getitem__
方法,那么开始把x[0]
、x[1]
、x[2]
一直到x[len(x) - 1]
不断地赋值给i
。 - 这里实际上执行的是
x.__getitem__(0)
、x.__getitem__(1)
、x.__getitem__(2)
一直到x.__getitem__(len(x) - 1)
。 - 最后执行到
x.__getitem__(len(x))
时,在__getitem__
方法内部要抛出一个IndexError
异常(StopIteration
也可以),从而终止迭代。
- 如果实现了
-
举个例子,该类实现了
__getitem__
方法,就可以用于for
循环迭代:
class Test:
def __init__(self):
self.val = [1, 2, 3]
def __getitem__(self, i):
if i < len(self.val):
return self.val[i]
else:
raise IndexError
x = Test()
for i in x:
print(i)
"""
1
2
3
"""
- 如果
__iter__
和__getitem__
两个方法都实现了,会优先调用x.__iter__()
方法。 - 如果两个方法都没有实现,会抛出
TypeError
异常,提示:TypeError: '类名' object is not iterable
完成于 2019.1.1