在 Python 中，迭代器是一种对象，它实现了迭代协议，即包含方法 __iter__() 和 __next__()。迭代器允许你遍历一个容器（如列表、元组等）中的所有元素，而无需显式地使用索引。理解迭代器的概念和使用可以帮助你编写更高效、可读性更强的代码。
迭代器的基本概念
- 迭代器对象：实现了
__iter__()和__next__()方法的对象。 - 可迭代对象：实现了
__iter__()方法，并返回一个迭代器的对象。例如，列表、元组、字典和集合都是可迭代对象。
创建迭代器
任何实现了 __iter__() 和 __next__() 方法的对象都是迭代器。可以通过创建一个类来实现自定义迭代器。
示例
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:
self.current += 1
return self.current - 1
else:
raise StopIteration
# 使用迭代器
it = MyIterator(1, 5)
for num in it:
print(num)
输出：
1
2
3
4
内置迭代器
Python 提供了许多内置函数和对象，它们是迭代器。例如，map、filter、zip 和 range 等。
示例
# map 迭代器
map_iterator = map(lambda x: x * 2, [1, 2, 3, 4])
print(list(map_iterator)) # 输出: [2, 4, 6, 8]
# zip 迭代器
zip_iterator = zip([1, 2, 3], ['a', 'b', 'c'])
print(list(zip_iterator)) # 输出: [(1, 'a'), (2, 'b'), (3, 'c')]
使用 iter() 和 next()
你可以使用 iter() 函数将一个可迭代对象转换为一个迭代器，并使用 next() 函数获取迭代器的下一个元素。
示例
# 创建一个列表迭代器
numbers = [1, 2, 3, 4]
it = iter(numbers)
print(next(it)) # 输出: 1
print(next(it)) # 输出: 2
print(next(it)) # 输出: 3
print(next(it)) # 输出: 4
# print(next(it)) # 会引发 StopIteration 异常
itertools 模块
Python 的 itertools 模块提供了许多用于创建和操作迭代器的工具。例如，count、cycle 和 repeat 等。
示例
import itertools
# count 迭代器
counter = itertools.count(start=10, step=2)
print(next(counter)) # 输出: 10
print(next(counter)) # 输出: 12
# cycle 迭代器
cycler = itertools.cycle(['A', 'B', 'C'])
print(next(cycler)) # 输出: A
print(next(cycler)) # 输出: B
# repeat 迭代器
repeater = itertools.repeat('Hello', 3)
print(list(repeater)) # 输出: ['Hello', 'Hello', 'Hello']
小结
- 迭代器：实现了
__iter__()和__next__()方法的对象，用于遍历元素。 - 可迭代对象：实现了
__iter__()方法，返回一个迭代器。 - 内置迭代器：如
map、filter、zip和range。 - 自定义迭代器：通过定义一个包含
__iter__()和__next__()方法的类来创建。 itertools模块：提供了丰富的迭代器工具。
通过理解和使用迭代器，可以编写更高效和优雅的代码，尤其是在处理大量数据时。