Python什么事迭代器?
在 Python 中,**迭代器**(Iterator)是一个实现了迭代协议的对象。这个协议包括两个方法:
1. **`__iter__()`**:返回迭代器对象本身,通常在此方法中初始化迭代所需的状态。
2. **`__next__()`**:返回序列中的下一个元素。如果没有更多元素可供迭代,抛出 `StopIteration` 异常。
在 Python 中,许多内建类型(如列表、元组、字典等)都支持迭代,因此我们可以使用 `for` 循环直接遍历它们,而不需要显式使用迭代器对象。但是,如果你希望自己实现一个迭代器,可以按照上述协议来实现。
### 例子 1:内建类型的迭代器
```python
# 使用内建类型的迭代器
my_list = [1, 2, 3, 4, 5]
iterator = iter(my_list) # 创建迭代器
# 使用 __next__() 来手动获取元素
print(next(iterator)) # 输出 1
print(next(iterator)) # 输出 2
print(next(iterator)) # 输出 3
# 当迭代器没有更多元素时,抛出 StopIteration 异常
try:
print(next(iterator)) # 输出 4
print(next(iterator)) # 输出 5
print(next(iterator)) # 抛出 StopIteration
except StopIteration:
print("迭代结束")
```
### 例子 2:自定义迭代器
我们也可以通过自定义一个类来实现迭代器。例如,创建一个简单的迭代器,能够返回一个给定范围内的数字。
```python
class MyIterator:
def __init__(self, start, end):
self.current = start
self.end = end
def __iter__(self):
return self # 迭代器对象本身
def __next__(self):
if self.current >= self.end:
raise StopIteration # 结束迭代
self.current += 1
return self.current - 1 # 返回当前元素
# 使用自定义的迭代器
iterator = MyIterator(0, 5)
for num in iterator:
print(num)
```
输出:
```
0
1
2
3
4
```
### 总结
- **迭代器**是实现了 `__iter__()` 和 `__next__()` 方法的对象,允许你逐个访问一个序列的元素。
- 可以通过内建的 `iter()` 函数将可迭代对象(如列表、字符串等)转化为迭代器。
- 使用 `next()` 函数可以手动从迭代器中获取下一个元素。