jacktangj 2019-10-30
可迭代的对象:常见的可以被for循环迭代的一些数据类型都是可迭代的对象,如列表,元组,字典,集合,字符串,生成器,range函数生成的数列等,从广泛的意义
上来说,这些对象都有一个内置的iter方法,且该方法可以返回一个迭代器对象,当用iter(可迭代对象)调用这个对象时,会返回一个迭代器对象(属于Iterator类)
在Python3中,实现了next方法和方法iter方法,并且这个iter这个方法返回了值的对象,就叫做迭代器或者迭代器对象。
判断可迭代对象和迭代器,从collections导入Iterable,Iterator,用isinstance判断
根据以上的介绍,我们可以按照这个思路实现自定义的迭代器
#迭代器对象类
class MyRangeIterator(object): def __init__(self, start, end): self.index = start self.end = end def __next__(self): if self.index < self.end: temp = self.index self.index += 1 return temp else: raise StopIteration()
#可迭代对象类
class MyRangeIterable(object): def __init__(self, start, end): self.start = start self.end = end def __iter__(self): # 该方法返回迭代器对象 return MyRangeIterator(self.start, self.end)
#1.直接采用for遍历可迭代对象:
for i in MyRangeIterable(1, 10): print(i)
#2.for底层的原理,for到底干了哪些事情:
#第一步,iter函数获取迭代器对象:
ret_iterator = iter(MyRangeIterable(1,10)) while True: try: x = next(ret_iterator) #或者ret_iterator.__next__,实际iter函数和next函数都会反射去执行对象的__next__和__iter__方法,道理一样 print(x) #如果迭代器没有返回值了就抛出异常,退出死循环 except StopIteration: break
print(‘----第二种实现迭代器的类---‘) class MyRangeIterable(object): def __init__(self, start, end): # self.start = start self.end = end self.index = start def __iter__(self): # 返回对象本身作为迭代器对象 return self def __next__(self): if self.index < self.end: temp = self.index self.index += 1 return temp else: raise StopIteration() my_range = MyRangeIterable(1,20) for i in my_range: print(i)