迭代器、迭代对象的 __iter__() 和 __next__()详解
1. 迭代器必须包含 __iter__()
和 __next__()
吗?
是的,Python 迭代器(Iterator)必须实现 __iter__()
和 __next__()
方法。但 __iter__()
其实可以返回 self
,它的作用只是为了让迭代器和可迭代对象兼容。
1.1 迭代器的定义
一个迭代器(Iterator) 必须:
- 实现
__iter__()
方法,返回自身self
。 - 实现
__next__()
方法,每次调用返回下一个值,遇到数据耗尽时抛出StopIteration
。
1.2 示例:自定义迭代器
class MyIterator:
def __init__(self, start, end):
self.current = start
self.end = end
def __iter__(self):
return self # 迭代器的 `__iter__()` 必须返回自身
def __next__(self):
if self.current >= self.end:
raise StopIteration # 迭代结束
value = self.current
self.current += 1
return value
# 创建迭代器对象
it = MyIterator(1, 5)
# 使用 `next()`
print(next(it)) # 1
print(next(it)) # 2
# 也可以用 `for` 循环
for num in MyIterator(1, 5):
print(num)
🔹 输出:
1
2
1
2
3
4
注意:
__iter__()
只是返回self
,让迭代器可以被iter()
调用(如iter(it)
)。__next__()
控制迭代逻辑,每次返回一个值,最后抛出StopIteration
。
1.3 为什么 __iter__()
必须返回 self
?
因为Python 规定 iter(iterator)
必须返回可用的迭代器: