Python的生成器和迭代器

Python生成器和迭代器

作者：王大为
时间：2016-10-24 11:00

一、生成器generator

分为以下两种：

1、列表生成器，形如：( i * i for i in range(10))
2、函数生成器，yield关键字

1、列表生成器

1.1、第一种情况（基本）
代码code：

结果result：

1.2、第二种情况（if判断）
代码code：

结果result：

1.3、第三种情况（循环for嵌套）
代码code：

结果result：

1.4、第四种情况（使用字典）
代码code：

结果result：

2、函数生成器
和函数一起使用：可以使函数中断并保存函数的中断状态，yield关键字，可以减少程序的阻塞

2.1、next()方法

yield 对应值：调用__next__()方法时返回的值
例如yield a，则__next__()方法时返回ayield + 后面的值：整个表达式为none（send函数传递的就是整个表达式的值进去）1、例如c = yield a，则c为None2、例如c = yield a，并且send('hello')，则c为'hello'

代码code：

def fib(s):n, a, b = 0, 0, 1while n < s:print('-------')yield bprint('hello')a, b = b, a + bn += 1return 'out of range'fib_gen = fib(5)
a = fib_gen.__next__()
print(a)

结果result：

2.2、异常StopIteration

异常的信息value就是函数生成器的return返回值信息

代码code：

def fib(s):n, a, b = 0, 0, 1while n < s:print('-------')yield ba, b = b, a + bn += 1return 'out of range'fib_gen = fib(5)
for i in range(6):try:print(fib_gen.__next__())except StopIteration as e:print(e.value)

结果result：

2.3、send()方法

生成器的send方法可以传递整个表达式（yield a）的值需要注意的是：生成器必须被运行后才可以使用send方法，表现为必须首先调用__next__方法，然后才可以使用send方法原因：就是python中，在执行fib_gen = fib(5)，并没有进入函数内部执行，这个时候，只是把fib_gen定义为了一个生成器；只有等fib_gen被执行了以后，才会启动生成器；所以在python代码里面，send比yield还早，这是有问题的需要首先添加fib_gen.__next__()方法

2.3.1、直接调用send方法
代码code：

def fib(s):n, a, b = 0, 0, 1while n < s:a = yield bprint(a)n += 1return 'out of range'fib_gen = fib(5)
fib_gen.send('hello')

结果result：

2.3.2、首先next()启动生成器，然后调用send方法
代码code：

def fib(s):n, a, b = 0, 0, 1while n < s:a = yield bprint(a)n += 1return 'out of range'fib_gen = fib(5)
fib_gen.__next__()
fib_gen.send('hello')

结果result：

二、迭代器iterator

1、迭代器的组成分类

一类是集合数据类型，例如：list，tuple，dict，set，str等
一类是generator对象，包括列表生成器和带yield的generator function

2、可迭代对象
定义：凡是可作用于for循环的对象都是iterable类型，可迭代对象

判断：使用isinstance()函数判断是否是可迭代对象
代码code：

from collections import Iterable
a = isinstance([], Iterable)
b = isinstance((), Iterable)
c = isinstance('', Iterable)
d = isinstance({}, Iterable)
e = isinstance(12, Iterable)
print(a)
print(b)
print(c)
print(d)
print(e)

结果result：

3、迭代器
定义：凡是可作用于next()的对象都是iterator类型，迭代器

4、可迭代对象——>迭代器的转化，iter()内置函数，将iterable可迭代对象转化为iterator迭代器
代码code：

from collections import Iterator
a = isinstance([], Iterator)
b = isinstance(iter(()), Iterator)print(a)
print(b)

结果result：

5、总结

生成器一定是迭代器，但是迭代器不一定是生成器（集合数据类型转化的迭代器）可作用于for循环的对象是可迭代对象可作用于next()的对象是迭代器可迭代对象可以通过iter()函数转化为迭代器