首页 > 编程语言 > 详细

Python-11-生成器

时间:2019-07-12 23:06:42      阅读:146      评论:0      收藏:0      [点我收藏+]

一、定义

可以理解为一种数据类型,这种数据类型自动实现了迭代器协议(其他数据类型需要调用__iter__方法),所以生成器就是一种迭代器。

二、生成器的两种形式

1. 生成器函数

使用yield代替return返回结果,yield语句一次返回一个结果,返回一个结果后,挂起函数的状态,下次继续执行。

def fib(max):
    n, a, b = 0, 0, 1
    while n < max:
        yield b
        a, b = b, a + b
        n = n + 1
    return ‘done‘


f = fib(10)
print(next(f))
print(next(f))
print(next(f))
print(next(f))
# 函数是顺序执行,遇到return语句或者最后一行函数语句就返回。而变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。
def odd():
    print(‘step 1‘)
    yield 1
    print(‘step 2‘)
    yield(3)
    print(‘step 3‘)
    yield(5)

>>> o = odd()
>>> next(o)
step 1
1
>>> next(o)
step 2
3
>>> next(o)
step 3
5
>>> next(o)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

2. 生成器表达式

# 三元表达式
name = ‘alex‘
res = ‘SB‘ if name == ‘alex‘ else ‘123‘
print(res)

SB

# 列表解析
l1 = [‘鸡蛋%s‘ % i for i in range(1, 10)]
l2 = [‘鸡蛋%s‘ % i for i in range(1, 10) if i > 5]
# l3 = [‘鸡蛋%s‘ % i for i in range(1, 10) if i > 5 else i] # 没有四元表达式
print(l1)
print(l2)

[‘鸡蛋1‘, ‘鸡蛋2‘, ‘鸡蛋3‘, ‘鸡蛋4‘, ‘鸡蛋5‘, ‘鸡蛋6‘, ‘鸡蛋7‘, ‘鸡蛋8‘, ‘鸡蛋9‘]
[‘鸡蛋6‘, ‘鸡蛋7‘, ‘鸡蛋8‘, ‘鸡蛋9‘]

# 生成器表达式
g = (‘鸡蛋%s‘ % i for i in range(1, 10) if i > 5)
print(next(g))
print(next(g))
print(next(g))
print(next(g))

鸡蛋6
鸡蛋7
鸡蛋8
鸡蛋9

Python-11-生成器

原文:https://www.cnblogs.com/lsf123456/p/11178544.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!