from collections import Iterator, Iterable
class MyListIterator(object): # 定义迭代器类,其是MyList可迭代对象的迭代器类
def __init__(self, data):
self.data = data # 上边界
self.now = 0 # 当前迭代值,初始为0
def __iter__(self):
return self # 返回该对象的迭代器类的实例;因为自己就是迭代器,所以返回self
def __next__(self): # 迭代器类必须实现的方法
while self.now < self.data:
self.now += 1
return self.now - 1 # 返回当前迭代值
raise StopIteration # 超出上边界,抛出异常
my_list = MyListIterator(5) # 得到一个可迭代对象
print(type(my_list)) # 返回该对象的类型 <class '__main__.MyListIterator'>
my_list_iter = iter(my_list) # 得到该对象的迭代器实例,iter函数在下面会详细解释 <class '__main__.MyListIterator'>
print(type(my_list_iter)) # <class '__main__.MyListIterator'>
for i in my_list: # 一次调用__next__方法
print(i) # 0,1,2,34
print(isinstance(my_list, Iterator)) # True
print(isinstance(my_list, Iterable)) # True
print(isinstance([], Iterable)) # True
print(isinstance([], Iterator)) # True
gene = (i for i in range(10)) # 创建一个生成器
print(gene, type(gene)) # <generator object <genexpr> at 0x0000024609D9B5C8> <class 'generator'>
print(isinstance(gene, Iterable)) # True
print(isinstance(gene, Iterator)) # True
总结:
- 如果一个对象拥有__iter__方法,其是可迭代对象
- 定义可迭代对象,必须实现__iter__方法;定义迭代器,必须实现__iter__和next方法
- 生成器是一种特殊的迭代器,生成器自动实现了“迭代器协议”(即__iter__和next方法),不需要再手动实现两方法(表现形式为函数里面使用了yield关键字)