系列文章目录
- 迭代器
- 迭代器的优点
一、迭代器
迭代是访问集合元素的一种方式,迭代器是一个可以记住遍历的位置的对象,迭代器对象从集合的第一个元素开始访问,知道所有的元素都被访问完结束,迭代器只能往前不能后退
可迭代的对象:我们已经知道可以对list,tuple,str等类型的数据使用for...in...的循环语法从其中依次拿到数据进行使用,我们把这样的过程称为遍历,也叫迭代
这里判断一个对象是否是可以迭代的方法时用Iterable,返回的值是True 就是可以迭代的。
如果一个对象本身是不可以迭代的对象,就可以使用__iter__方法,把它变成可以迭代的对象
下面就是实现一个可以迭代的对象操作:
简单一点讲,后面的for循环就是通过这个迭代器来取值的
操作:
代码:
import time
from collections.abc import Iterable
from collections.abc import Iterator
class Classmate(object):
def __init__(self):
self.names = list()
def __add__(self, name):
self.names.append(name)
def __iter__(self):
"""如果想要一个对象称为一个 可以迭代的对象,既可以使用for,那么必须实现__iter__方法"""
return ClassIterator(self)
class ClassIterator(object):
def __init__(self, obj):
self.obj = obj
self.current_num = 0
def __iter__(self):
pass
def __next__(self):
if self.current_num < len(self.obj.names):
result = self.obj.names[self.current_num]
self.current_num += 1
return result
else:
raise StopIteration
classmate = Classmate()
classmate.__add__("张三")
classmate.__add__("李四")
classmate.__add__("王五")
print("判断classmate是否是可以迭代的对象:", isinstance(classmate, Iterable))
# 输出结果是True 就是可以迭代的
classmate_iterator = iter(classmate)
print("判断classmate_iterator是否是迭代器:", isinstance(classmate, Iterable))
iter(classmate)
for name in classmate:
print(name)
time.sleep(1
输出结果:
二、迭代器的优点
迭代器的优点:和返回列表的方式相比较,只占同极小的内存空间。最终存储是生成数据的方式,而不是存储这个数据的结果
举个例子:数学中有个著名的斐波拉契数列,数列中第一个数为0.第二个数为1,其中的每一个数都可由前两个数相加得到:
0,1,1,2,3,5,8,13,21,34,....
现在就可以用迭代器来实现,每次迭代都通过数学计算来生成下一个数
首先,先操作一个不使用迭代器的操作:
输出结果:
不使用迭代器的话,重复的代码就相当于的多
再简化一下,使用while循环
下面就是使用迭代器的操作:
代码:
class Fibonacci(object):
def __init__(self, all_num):
self.all_num = all_num
self.current_num = 0
self.a = 0
self.b = 1
def __iter__(self):
return self
def __next__(self):
if self.current_num < self.all_num:
result = self.a
self.a, self.b = self.b , self.a + self.b
self.current_num += 1
return result
else:
raise StopIteration
fibo = Fibonacci(10)
for num in fibo:
print(num)
输出结果:
第一个使用while循环和使用迭代器相比较,迭代器从头到尾没有使用过列表,就能运算出所有斐波拉契数列的结果。
虽然看着目前感觉迭代器比较复杂一些,但是要是计算的数值很多的时候,列表就会占用很大的空间来保存列表所产生的的值,而迭代器只是需要next方法那一段代码就能够保存上千万个值