yield
把函数变成生成器函数,成为可迭代对象,可以用for循环遍历
#比如,著名的斐波拉契数列(Fibonacci),除第一个和第二个数外,任意一个数都可由前两个数相加得到:
1, 1, 2, 3, 5, 8, 13, 21, 34, ...
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(7)
print(i)
#输出 1 1 2 3 5 8 13
#杨辉三角生成器函数(generator)
def tri():
list = [1]
while True:
yield list
list = [1] + [list[i]+list[i+1] for i in range(len(list)-1)] +[1]
n=0
result = []
for t in tri():
print(t)
result.append(t)
n+=1
if n==10:
break
#输出
补充
列表生成器
列表生成式即List Comprehensions,是Python内置的非常简单却强大的可以用来创建list的生成式。
>>>list(range(1, 11))
>>>[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> [x * x for x in range(1, 11)]
>>>[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
生成器
通过列表生成式,我们可以直接创建一个列表。但是,受到内存限制,列表容量肯定是有限的。而且,创建一个包含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] #list,dict,str是迭代器但不是生成器
>>> g = (x * x for x in range(10))
>>> g
<generator object <genexpr> at 0x1022ef630>
Tip:
凡是可作用于for循环的对象都是Iterable类型(可迭代对象);
凡是可作用于next()函数的对象都是Iterator类型(迭代器),它们表示一个惰性计算的序列;
集合数据类型如list、dict、str等是Iterable但不是Iterator,不过可以通过iter()函数获得一个Iterator对象。
Python的for循环本质上就是通过不断调用next()函数实现的
参照廖雪峰老师网站
其他资料;