生成器:generator,主要是处理无限大的序列,但不希望采用传统的生成再存储的方式,因为这样极度占用内存与存储空间,因此可以看做是个懒人函数,需要的时候再调用,未调用时可以说只存储一个生成的表达式。
如
s = (x*x for x in range(10))
print(s)
这里执行结果并不是所期望得到的列表,而是 <generator object <genexpr> at 0x00000273C3773C78> ,如果需要得到列表,我们需要加上:
for i in range(10):
print(next(i))
注意这里有next函数,如果执行完毕再继续next,就会抛出Iteration Error。为了规避,可以用for循环如下语句:
for i in s:
print(i)
生成器关键字 yield, 如斐波那契数列:
def fib(max):
n,a,b =0,1,1
while n < max:
yield a
a,b =b,a+b
n = n+1
return 'done'
a = fib(10)
for i in a:
print(i)
更进一步,尝试生成杨辉三角:(最早期写了个特别锉的代码,鄙视一下自己: https://blog.csdn.net/weixin_44521703/article/details/91043327)
def triangles():
N = [1]
while True:
yield N
N = [1] + [N[i] + N[i+1] for i in range(len(N) - 1) ]+ [1]
n = 0
results = []
for t in triangles():
print(t)
results.append(t)
n = n + 1
if n == 10:
break
生成结果:
[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]
(2019年8月3日)
今天看到一个很好的思路:
比如上一行是[1,2,1],下一行可以这么构造,造两个列表,在其前面与后面分别加个0,即做成[0,1,2,1]与[1,2,1,0],两个列表主元素相加就是[1,3,3,1]
这个思路实际也很有技巧的,实现算法如下:
def generate(numRows):
r = [[1]]
for i in range(1, numRows):
r.append(list(map(lambda x, y: x + y, [0] + r[-1], r[-1] + [0])))
return r[:numRows]
a = generate(10)
for i in a:
print(i)
输出结果是:
[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]