Python迭代器
1. 迭代器对象
自定义迭代器对象: 在类里面定义
__iter__和__next__方法创建的对象就是迭代器对象
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
from collections import Iterable from collections import Iterator # 自定义可迭代对象: 在类里面定义__iter__方法创建的对象就是可迭代对象 class MyList(object): def __init__(self): self.my_list = list() # 添加指定元素 def append_item(self, item): self.my_list.append(item) def __iter__(self): # 可迭代对象的本质：遍历可迭代对象的时候其实获取的是可迭代对象的迭代器， 然后通过迭代器获取对象中的数据 my_iterator = MyIterator(self.my_list) return my_iterator # 自定义迭代器对象: 在类里面定义__iter__和__next__方法创建的对象就是迭代器对象 class MyIterator(object): def __init__(self, my_list): self.my_list = my_list # 记录当前获取数据的下标 self.current_index = 0 # 判断当前对象是否是迭代器 result = isinstance(self, Iterator) print("MyIterator创建的对象是否是迭代器:", result) def __iter__(self): return self # 获取迭代器中下一个值 def __next__(self): if self.current_index < len(self.my_list): self.current_index += 1 return self.my_list[self.current_index - 1] else: # 数据取完了，需要抛出一个停止迭代的异常 raise StopIteration my_list = MyList() my_list.append_item(1) my_list.append_item(2) result = isinstance(my_list, Iterable) print(result) for value in my_list: print(value) |
运行结果:
|
1 2 3 4 5 |
True MyIterator创建的对象是否是迭代器: True 1 2 |
2. iter()函数与next()函数
- iter函数: 获取可迭代对象的迭代器，会调用可迭代对象身上的
__iter__方法 - next函数: 获取迭代器中下一个值，会调用迭代器对象身上的
__next__方法
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
# 自定义可迭代对象: 在类里面定义__iter__方法创建的对象就是可迭代对象 class MyList(object): def __init__(self): self.my_list = list() # 添加指定元素 def append_item(self, item): self.my_list.append(item) def __iter__(self): # 可迭代对象的本质：遍历可迭代对象的时候其实获取的是可迭代对象的迭代器， 然后通过迭代器获取对象中的数据 my_iterator = MyIterator(self.my_list) return my_iterator # 自定义迭代器对象: 在类里面定义__iter__和__next__方法创建的对象就是迭代器对象 # 迭代器是记录当前数据的位置以便获取下一个位置的值 class MyIterator(object): def __init__(self, my_list): self.my_list = my_list # 记录当前获取数据的下标 self.current_index = 0 def __iter__(self): return self # 获取迭代器中下一个值 def __next__(self): if self.current_index < len(self.my_list): self.current_index += 1 return self.my_list[self.current_index - 1] else: # 数据取完了，需要抛出一个停止迭代的异常 raise StopIteration # 创建了一个自定义的可迭代对象 my_list = MyList() my_list.append_item(1) my_list.append_item(2) # 获取可迭代对象的迭代器 my_iterator = iter(my_list) print(my_iterator) # 获取迭代器中下一个值 # value = next(my_iterator) # print(value) # 循环通过迭代器获取数据 while True: try: value = next(my_iterator) print(value) except StopIteration as e: break |
3. for循环的本质
遍历的是可迭代对象
- for item in Iterable 循环的本质就是先通过iter()函数获取可迭代对象Iterable的迭代器，然后对获取到的迭代器不断调用next()方法来获取下一个值并将其赋值给item，当遇到StopIteration的异常后循环结束。
遍历的是迭代器
- for item in Iterator 循环的迭代器，不断调用next()方法来获取下一个值并将其赋值给item，当遇到StopIteration的异常后循环结束。
4. 迭代器的应用场景
我们发现迭代器最核心的功能就是可以通过next()函数的调用来返回下一个数据值。如果每次返回的数据值不是在一个已有的数据集合中读取的，而是通过程序按照一定的规律计算生成的，那么也就意味着可以不用再依赖一个已有的数据集合，也就是说不用再将所有要迭代的数据都一次性缓存下来供后续依次读取，这样可以节省大量的存储（内存）空间。
举个例子，比如，数学中有个著名的斐波拉契数列（Fibonacci），数列中第一个数为0，第二个数为1，其后的每一个数都可由前两个数相加得到：
0, 1, 1, 2, 3, 5, 8, 13, 21, 34, …
现在我们想要通过for…in…循环来遍历迭代斐波那契数列中的前n个数。那么这个斐波那契数列我们就可以用迭代器来实现，每次迭代都通过数学计算来生成下一个数。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
class Fibonacci(object): def __init__(self, num): # num:表示生成多少fibonacci数字 self.num = num # 记录fibonacci前两个值 self.a = 0 self.b = 1 # 记录当前生成数字的索引 self.current_index = 0 def __iter__(self): return self def __next__(self): if self.current_index < self.num: result = self.a self.a, self.b = self.b, self.a + self.b self.current_index += 1 return result else: raise StopIteration fib = Fibonacci(5) # value = next(fib) # print(value) for value in fib: print(value) |
执行结果:
|
1 2 3 4 5 6 |
0 1 1 2 3 |
tips：
迭代器的作用就是是记录当前数据的位置以便获取下一个位置的值
