美文网首页
Python学习之路(高级特性之生成器generator)

Python学习之路(高级特性之生成器generator)

作者: 55lover | 来源:发表于2017-12-22 10:32 被阅读0次

高级特性之生成器

前面我们学习了高级特性的列表生成式 ,list [] 但是list 数据量如果过大 那么在内存中就占了大量空间 是一个不好的现象,
现在我们改用生成器 即 tuple 用()来包起来 不用[]
返回的对象 就是一个生成器generator (可迭代的对象)
如果需要拿取数据 只需用next(索引)

# 比如:
g = (x * x for x in range(11))
print(g) # 返回generator对象
# print(next(g))
# print(next(g))
# print(next(g))
# print(next(g))
# print(next(g))

L1 = [ x for i, x in enumerate(g) if i<5] # 遍历generator 对象  [0, 1, 4, 9, 16]
print(L1)
L2 = [ str(i)+'='+str(x) for i, x in enumerate(g) if i<5] # 遍历generator 对象 ['0=0', '1=1', '2=4', '3=9', '4=16']
print(L2)

练习

著名的斐波拉契数列(Fibonacci),除第一个和第二个数外,任意一个数都可由前两个数相加得到:
1, 1, 2, 3, 5, 8, 13, 21, 34, ...

def fib(max): 
    n, a, b = 0, 0, 1
    while n < max:
        yield b # 返回的是generator对象 所以 可以迭代
        a, b = b, a + b
        n += 1
    return 'done'
print(fib(6))

print([x for x in fib(12)])

generator和函数的执行流程不一样。函数是顺序执行,遇到return语句或者最后一行函数语句就返回。而变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。

def odd():
    print('step 1')
    yield 1
    print('step 2')
    yield 2
    print('step 3')
    yield 3
o = odd()
print(next(o))
print(next(o))
print(next(o))
# 依次执行odd() 并返回

杨辉三角定义如下:
python实现杨辉三角原理:

def yh():
    L = [1]
    while True:
        yield L
        L.append(0)
        L = [L[i - 1] + L[i] for i in range(len(L))]
n = 0
results = []
for t in yh():
    results.append(t)
    n = n + 1
    print(t)  # 打印结果
    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]

小结:

generator是非常强大的工具,在Python中,可以简单地把列表生成式改成generator,也可以通过函数实现复杂逻辑的generator。

要理解generator的工作原理,它是在for循环的过程中不断计算出下一个元素,并在适当的条件结束for循环。对于函数改成的generator来说,遇到return语句或者执行到函数体最后一行语句,就是结束generator的指令,for循环随之结束。

请注意区分普通函数和generator函数,普通函数调用直接返回结果:

# >>> r = abs(6)
# >>> r
# 6
# generator函数的“调用”实际返回一个generator对象:

# >>> g = fib(6)
# >>> g
# <generator object fib at 0x1022ef948>

关注一波!喜欢一波!本人是前端菜鸟,正在做自己的个人博客邓鹏的博客,欢迎来交流学习, 使用的技术 vue + koa2 + mysql + php + nginx!

相关文章

网友评论

      本文标题:Python学习之路(高级特性之生成器generator)

      本文链接:https://www.haomeiwen.com/subject/fmdtgxtx.html