迭代器
1、 从一个遍历问题说起
python中的字符串可以使用for循环遍历吗?为什么?
python中的整数可以使用for循环遍历吗?为什么?
2、 什么是迭代对象?什么是迭代器?
字符串中包含一个魔术方法:iter()
(1) 什么是可迭代对象
一个对象/类型中包含__iter__()魔术方法,这个对象/类型,就是可迭代对象!
(2) 什么是迭代器
迭代器就是一个对象中包含__next__()魔术方法的对象。
① 生成器就是一种特殊的迭代器
② 迭代器的获取和使用
3、 自定义迭代器语法
① 作为一个可迭代对象,必须包含__iter__()方法
② 当前类型也作为迭代器对象,必须包含__next__()方法
"""
自定义迭代器对象
"""
class Person:
"""自定义可迭代对象"""
def __init__(self, fav=("王者荣耀", "魂斗罗", "守望先锋")):
self.fav = fav
self.index = 0
def __iter__(self):
"""返回迭代器对象:这里把当前对象当成迭代器对象"""
return self
def __next__(self):
"""获取下一个数据"""
d = self.fav[self.index]
self.index += 1
return d
p = Person()
for x in p:
print(x)
迭代器案例操作:在线人数计数器:
"""
迭代器实现计数器
"""
class OnlineCount:
def __init__(self):
self.index = 0
def __iter__(self):
return self
def __next__(self):
self.index += 1
return self.index
# 获取可迭代对象
oc = OnlineCount()
# 通过可的迭代对象直接获取下一个数据
print(next(oc))
print(next(oc))
print(next(oc))
print(next(oc))