关于迭代器的一些知识点
知识点:可迭代对象
关于可迭代对象的定义:
一般来说,我们使用的str,list,tuple,dict等的这些数据容器都是可迭代对象。可迭代对象都有可以特性,就是可以用for循环来遍历。
程序执行时,每一次对过程的重复称为一次“迭代”,每一次迭代的结果都会成为下一次迭代的初始值。
定义:具有__iter__ 函数的对象,可以返回迭代器对象
作用:可以参与for循环
可迭代对象的标志:具有__iter__(),返回值是迭代器。
迭代器:
标志:具有__next__(),返回值是聚合对象的元素
作用:可以迭代
语法
– 创建:
class 可迭代对象名称:
def _ iter_(self):
return 迭代器
– 使用:
for 变量名 in 可迭代对象:
语句
3. 原理:
迭代器 = 可迭代对象._ iter_()
while True:
try:
print(迭代器._ next_())
except StopIteration:
break
具体例子:
需求:让技能管理器对象可以参与for循环
class SkillManager:
"""
技能管理器
具有__iter__方法,可迭代对象.
"""
def __init__(self):
self.__all_skill = []
def add_skill(self, skill):
self.__all_skill.append(skill)
def __iter__(self):
# 创建技能迭代器对象(传递需要迭代的数据)
return SkillIterator(self.__all_skill)
需要注意的地方:
def iter(self):
return SkillIterator(self.__all_skill)
我们的创建的是迭代器的对象,返回的的就是需要迭代的数据,这也是与SkillManager与下面SkillIteratior所创建的联系。
class SkillIterator:
"""
技能迭代器
"""
def __init__(self, data):
self.__target = data
self.__index = -1 # 返回数据时,会先自增1
def __next__(self):
self.__index += 1
# 当前需要获取的索引 比最大索引还大
if self.__index > len(self.__target) - 1:
raise StopIteration
return self.__target[self.__index]
manager = SkillManager() #创建对象
manager.add_skill("九阳神功")
manager.add_skill("乾坤大挪移")
manager.add_skill("加藤之手")
到了这里,很多人会问两个类之间的联系究竟要怎么创建,当时我也懵了很久。甚至一度认为SkillIteratior是SkillManager的派生类,后来我发现他们两个之间的调用用继承关系是无法实现,而且这种想法是大错特错的。所以,在这里再次强调,SkillManager和SkillIteratior他们是相互独立的,不存在继承的关系。所以上面SkillManager.__iter__的作用是把需要迭代的数据传给了迭代器。
下面,我们再进行测试:
for item in manager:
print(item)