python学习笔记–生成器
- 通过列表生成式,我们可以直接创建一个列表。但是,受到内存限制,列表容量肯定是有限的。而且,创建一个包含100万个元素的列表,不仅占用很大的存储空间,如果我们仅仅需要访问前面几个元素,那后面绝大多数元素占用的空间都白白浪费了。
- 所以,如果列表元素可以按照某种算法推算出来,那我们是否可以在循环的过程中不断推算出后续的元素呢?这样就不必创建完整的list,从而节省大量的空间。在Python中,这种一边循环一边计算的机制,称为生成器:generator。
- 要创建一个generator,有很多种方法。第一种方法很简单,只要把一个列表生成式的[]改成(),就创建了一个generator:
>>> L = [x * x for x in range(10)]
>>> L
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> g = (x * x for x in range(10))
>>> g
<generator object <genexpr> at 0x01D1C270>
下面用生成器生成杨辉三角
1
/ \
1 1
/ \ / \
1 2 1
/ \ / \ / \
1 3 3 1
/ \ / \ / \ / \
1 4 6 4 1
/ \ / \ / \ / \ / \
1 5 10 10 5 1
def triangles(): #杨辉三角
L=[1]
while True:
yield L
L=[L[i]+L[i+1] for i in range(len(L)-1)]
L.insert(0,1)
L.append(1)
L=[L[i]+L[i+1] for i in range(len(L)-1)]
- 上式并不会下标索引溢出,因为对于for循环,当range()里面为(0,0)时,i = None,此时循环并不会运行,运行结果则位L = []
- 测试代码如下:
a = 0
for i in range(0,0):
a = 1
print('--')
if a == 0:
print('for 循环中迭代为(0,0)时不执行')
- 运行结果为:
for 循环中迭代为0时不执行
- 生成器代码测试:
n = 0
results = []
for t in triangles():
print(t)
results.append(t)
n = n + 1
if n == 10:
break
if results == [
[1],
[1, 1],
[1, 2, 1],
[1, 3, 3, 1],
[1, 4, 6, 4, 1],
[1, 5, 10, 10, 5, 1],
[1, 6, 15, 20, 15, 6, 1],
[1, 7, 21, 35, 35, 21, 7, 1],
[1, 8, 28, 56, 70, 56, 28, 8, 1],
[1, 9, 36, 84, 126, 126, 84, 36, 9, 1]
]:
print('测试通过!')
else:
print('测试失败!')
学习网站:廖雪峰的官方博客–生成器