Python 列表生成式和生成器?可迭代和迭代器?
列表生成式和生成器?
列表生成式,就是一个能够生成列表的推导式,用[]括起来,里面是一个包含for循环的推导式,形式一般如下:
[x * x for x in range(1, 11) if x % 2 == 0]
意思是从1-10遍历,选取偶数数字然后平方组成新的列表。
生成器,generator,两种写法,第一种把上面的列表生成式的中括号[]改成小括号(),如下:
g = (x * x for x in range(10))
需要注意这样得到的不是一个列表也不是元组,也不是1-10每个数平方后的数据,而是一个生成器对象,它又么用呢?
使用列表生成式是一下子把所有元素都计算出来,比如说我们最终要100万个元素,列表生成式一下子计算就得到了100万个元素,但是就目前而言我只用到前10个后面的元素以后才可能用,那一下子计算出100万个又慢有占空间。
所以生成器就可以按需计算,即我们只需要告诉它计算推导的规则就行,想要得到下一个元素,我们就调用next()方法按照规则计算获取下一个元素,当然也可以用for循环获取前n个元素,所以它有一种惰性计算的性质。
比如上面代码中的生成器,可以使用next()方法不断获取下一个元素:
g = (x * x for x in range(10))
next(g) # 0
next(g) # 1
next(g) # 4
也可以使用for循环:
g = (x * x for x in range(10))
for n in g:
print(n)
# 0
# 1
# 4
# 9
...
# 81
第二种方式就是通过函数告诉它计算规则,但是函数毕竟是函数,怎么变成生成器呢,只要每次计算出数据用yield关键字返回就行了,比如计算斐波那契数量:
def fib(max):
n, a, b = 0, 0, 1
while n < max:
yield b # 计算出b并返回,下次执行从上次终止的地方开始而不是重新开始
a, b = b, a + b
n = n + 1
return 'done'
也是可以通过next()或者for循环去遍历,比如使用next()方法:
g = fib(6) # 创建一个生成6个斐波那契数的生成器
next(g)
# 1
next(g)
# 1
next(g)
# 2
next(g)
# 3
next(g)
# 5
next(g)
# 8
next(g)
# Traceback (most recent call last):
# File "<stdin>", line 1, in <module>
# StopIteration
发现每次执行next()方法,它是从上次停止的地方继续执行,而不会像函数一样从头开始,最后结束如果继续调用next()方法会返回StopIteration
异常。
使用for循环同理:
for n in fib(6):
print(n)
# 1
# 1
# 2
# 3
# 5
# 8
可迭代和迭代器?
可以直接作用于for
循环的数据类型有以下几种:
一类是集合数据类型,如list
、tuple
、dict
、set
、str
等;
一类是generator
,包括生成器和带yield
的generator function。
凡是可作用于for
循环的对象都是可迭代的,叫Iterable
类型;
凡是可作用于next()
函数的对象都是迭代器,叫Iterator
类型,它们表示一个惰性计算的序列;
集合数据类型如list
、dict
、str
等是Iterable
但不是Iterator
,不过可以通过iter()
函数获得一个Iterator
对象。
Python的Iterator
对象也就是迭代器,表示的是一个数据流,Iterator对象可以被next()
函数调用并不断返回下一个数据,直到没有数据时抛出StopIteration
错误。可以把这个数据流看做是一个有序序列,但我们却不能提前知道序列的长度,只能不断通过next()
函数实现按需计算下一个数据,所以Iterator
的计算是惰性的,只有在需要返回下一个数据时它才会计算。
Iterator
甚至可以表示一个无限大的数据流,例如全体自然数。而使用list是永远不可能存储全体自然数的。