Python - __iter__的内部工作原理

Question

我有以下类:

class RandomNumbers:
def __init__(self, length, *, range_min=0, range_max=10):
self.length = length
self.range_min = range_min
self.range_max = range_max
self.i = 0
def __len__(self):
return self.length
def __next__(self):
if self.i >= self.length:
raise StopIteration
number = random.randint(self.range_min, self.range_max)
self.i += 1
return number
def __iter__(self):
print("I was called")
return self

这允许我使用for循环:

for number in RandomNumbers(10):
print(number)

这工作。当我注释掉__iter__时，我得到以下错误:

TypeError: 'RandomNumbers' object is not iterable

好的，到目前为止一切顺利。我知道我需要__iter__方法，但为什么需要它，当它实际上只返回self?

Answer 1

您的for循环可以重写为以下代码

it = iter(RandomNumbers(10))
while True:
try:
number = next(it)
except StopIteration:
break
else:
# Body of the for loop
print(number)

你的类需要实现__iter__方法来用一个迭代器响应iter()函数，在你的例子中是self，因为你的类实现了__next__，使它成为一个迭代器