python – 内置类型 之 迭代器类型
目录
一、迭代器iterator
1.理解迭代器
2.迭代器函数iter和next
3.迭代器运用案例
二、生成器generator
1.理解生成器
2.创建生成器
3.yield函数
一、迭代器iterator
1.理解迭代器
(1)什么是迭代器
迭代器也称迭代对象，指可for .. in ..等语句进行循环，这样的过程叫迭代，该对象称为迭代对象
迭代器用于迭代数据，记录每次访问的数据，以及返回下一个数据，用__next__取下一个指向的值
(2)可迭代对象有哪些
列表，元祖，字典，字符串，set集合等都是可迭代对象
(3)通过isinstance()来判断一个对象是否是迭代对象
2.迭代器函数iter和next
(1)iter(iterable)
从可迭代对象中返回一个迭代器,iterable必须是能提供一个迭代器的对象
(2)next(iterator)
从迭代器iterator中获取下一了记录,如果无法获取下一条记录,则触发stoptrerator异常
注意:迭代器只能往前取值,不会后退
3.迭代器运用案例
(1)创建迭代器
l = [1,3,5,7]
it = iter(l)
next(it) #1
next(it) #3
next(it) #5
next(it) #7
next(it) #StopIterable 异常
(2)用迭代器轮循列表
l = [2,3,5,7]
it = iter(l)
while True:
try:
x = next(it) #获取一个数据并绑定到x
print(x) #2,3,5,7
except StopIteration:
break
(3)自定义可迭代的类
在类中添加了__iter__方法即取得迭代器
在通过__next__方法指出所有的数据
class Fab(object):
def __init__(self, max):
self.max = max
self.n, self.a, self.b = 0, 0, 1
def __iter__(self):
“””返回一个迭代器”””
return self
def __next__(self):
“””返回迭代器下一个指向的值”””
if self.n < self.max:
r = self.b
self.a, self.b = self.b, self.a + self.b
self.n = self.n + 1
return r
else:
raise StopIteration()
if __name__==’__main__’:
for key in Fab(5):
print(key)
二、生成器generator
1.理解生成器
生成器有什么作用： 省内存
生成器一次只能返回一个值，将大大减小占用内存，而且允许调用函数快速的处理前几个返回值，因此生成器看起来像函数，但表现得却像迭代器
2.创建生成器
(1)把列表的[]改为()就创建generator
lis = [x*x for x in range(10)]
print(lis)
#[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
#生成器
generator_ex = (x*x for x in range(10))
print(generator_ex)
# <generator object <genexpr> at 0x000002A4CBF9EBA0>
使用next(generator_ex)函数获得generator的下一个返回值
print(next(generator_ex))
print(next(generator_ex))
print(next(generator_ex))
结果：
0
1
4
每次调用next()就返回generator的下一个值，多次调用next()直到*后一个元素，若再调用next()则抛出StopIteration的错误
所以正确的方法是使用for循环，因为generator也是可迭代对象，所以我们创建一个generator后，基本上永远不会调用next()，而是通过for循环来迭代
#生成器
generator_ex = (x*x for x in range(10))
for i in generator_ex:
print(i)
3.yield函数
yield 函数创建生成器generator。yield相当于 return 返回一个值，并且记住这个返回的位置，下次迭代时，代码从yield的下一条语句开始执行。
send()和next()一样，都能让生成器继续往下走一步（下次遇到yield停），但send()能传一个值，这个值作为yield表达式整体的结果
例子:
def fib(max):
n,a,b =0,0,1
while n < max:
yield b
a,b =b,a+b
n = n+1
return ‘done’
for i in fib(6):
print(i)
结果：
1
1
2
3
5
8
def fib(max):
n,a,b =0,0,1
while n < max:
yield b
a,b =b,a+b
n = n+1
return ‘done’
g = fib(6)
while True:
try:
x = next(g)
print(‘generator: ‘,x)
except StopIteration as e:
print(“生成器返回值：”,e.value)
break
结果：
generator: 1
generator: 1
generator: 2
generator: 3
generator: 5
generator: 8
生成器返回值： done