前言
这节内容主要介绍迭代器,看完文章你可以了解到什么是可迭代对象、啥是迭代器、如何自定义迭代器、使用迭代器的优势。
迭代器
迭代(iterate)意味着重复多次,就像循环那样。工作中我们一定使用for循环迭代过列表和字典,但实际上也可迭代其他对象:实现了方法__iter__
的对象。方法__iter__
返回一个迭代器,它是包含方法__next__
的对象,而调用这个方法时可不提供任何参数。当你调用方法__next__
时,迭代器应返回其下一个值。如果迭代器没有可供返回的值,应引发StopIteration异常。你还可使用内置的便利函数next。
__iter__
用于定义一个可迭代对象(iterable)。通过实现 __iter__
方法,我们可以使自定义的对象能够被 for
循环等迭代器相关的操作使用。
p ython中哪些是可迭代对象呢?我们可以测试一下
def is_iterable(param):
try:
iter(param)
return True
except TypeError:
return False
params = [
1234,
'1234',
[1, 2, 3, 4],
set([1, 2, 3, 4]),
{1: 1, 2: 2, 3: 3, 4: 4},
(1, 2, 3, 4)
]
for param in params:
print(f'{param} is iterable? {is_iterable(param)}')
'''
1234 is iterable? False
1234 is iterable? True
[1, 2, 3, 4] is iterable? True
{1, 2, 3, 4} is iterable? True
{1: 1, 2: 2, 3: 3, 4: 4} is iterable? True
(1, 2, 3, 4) is iterable? True
'''
可以看到列表、元组、字典、集合、字符串都是可迭代对象,查看源码,发现他们都实现了__iter__
方法
def __iter__(self, *args, **kwargs): # real signature unknown
""" Implement iter(self). """
pass
方法__iter__
返回一个迭代器,它是包含方法__next__
的对象,而调用这个方法时可不提供任何参数。当你调用方法__next__
时,迭代器应返回其下一个值。
实践
示例一
迭代器实现斐波那契数列
class Fibs:
def __init__(self):
s