~~函数进阶(二):生成器~~

进击のpython


生成器


上来说个这,就有点抽象了!

我们先整点活儿

宁,准备好了吗?

直接相位猛冲!


  • 列表生成器

    需求来了,老弟!我有一个数组

    a = [1, 2, 3, 4, 5, 6, 7, 8, 9]

    我想把列表里每个元素加一,怎么办呢?

    a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
    b = []
    for i in a:
        b.append(i+1)
    
    print()
    
    b

    还有吗?

    a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
    
    for index, i in enumerate(a):
        a[index] += 1
    print(a)

    还有吗?

    a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
    
    a = list(map(lambda x:x+1,a))
    print(a)

    还有吗?

    a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
    
    a = [i + 1 for i in a]
    print(a)

    最后一种的方式就是列表生成式(是不是很像三目运算??)


  • 生成器

    生成器,比如说我要是输出1~100

    怎么输出呢?

    for i in range(1,101):
        print(i)

    这大家都知道(不知道的滚回去看基础知识去,别在这捣乱)

    那有没有想过,range(1,100) 是什么呢?

    在py2.x系统下,我们可以看出来

    range(1,101)实际上是生成了一个从1~100的数字的列表

    那我要是有个这种句子呢?

    for i in range(1, 101):
        if i < 20:
            print(i)

    可以看出来我只需要20个数字,但是range给我生成了一个带有很多元素的列表

    即使里面大部分我都用不上,但是还是生成了。(while也行,别在这艮)

    那就会出现一个浪费空间的问题

    那我就要想了,能不能先生成一个我先用着,然后用完了再传下一个

    这个东西 就是 生成器

    做呢?有两种方式!


    将列表推导式的中括号变成小括号

    f = (x + 1 for x in range(10))
    
    print(f)                        # <generator object <genexpr> at 0x05681EF0>

    只是生成了生成规则,还没有产生数值

    那我要怎么产生数值呢?

    利用函数next()

    f = (x + 1 for x in range(10))
    
    print(next(f)) # 1
    print(next(f)) # 2
    print(next(f)) # 3
    print(next(f)) # 4
    print(next(f)) # 5 
    print(next(f)) # 6

    如果值取完了就会报错:

    要是想再使用这个生成器,那你就需要重新调用


    用循环来取

    这样就不会报错,会把值全都取出来

    f = (x + 1 for x in range(10))
    
    for i in f:
        print(i)

  • 函数生成器

    在做函数生成器之前,我们先试试输出100以内的斐波那契数列

    # 1 1 2 3 5 8 13 ... ...
    
    li = [1, 1]
    
    for i in range(2, 101):
        if li[i - 2] + li[i - 1] < 101:
            li.append(li[i - 2] + li[i - 1])
        else:
            break
    print(li)

    或者也可以这么写:

    # 0 1 1 2 3 5 8 13 ... ...
    
    a = 0
    b = 1
    count = 0
    while count < 20:
        tep = a
        a = b
        b = tep + b
        print(a,end=" ")
        count += 1

    那要是把这个东西改成函数怎么写??

    那很简单啊!

    # 0 1 1 2 3 5 8 13 ... ...
    
    def fib(n):
        a = 0
        b = 1
        count = 0
        while count < n:
            tep = a
            a = b
            b = tep + b
            print(a, end=" ")
            count += 1

    那函数生成器呢???更简单了

    看到print()没有?把 print()⇨ yield()

    大功告成!

    # 0 1 1 2 3 5 8 13 ... ...
    
    def fib(n):
        a = 0
        b = 1
        count = 0
        while count < n:
            tep = a
            a = b
            b = tep + b
            # print(a, end=" ")
            yield a
            count += 1
    
    
    print(fib(20))

    拿到了什么结果??(<generator object fib at 0x033A1EF0>)

    这是不是个生成器?(当然是!那么大个generator看不到啊)

    yield 暂停 你可以把它理解成 return 但是他后面的语句还是会执行的函数没有退出

    生成器是不是有 next() 函数啊,试试用 next() 打印这个结果

    有啥用???

    可太有用了!

    你没发现这个yield将函数进行中断操作了吗?

    什么意思呢?就是你可以执行着函数,然后停住他,执行点别的,再执行函数

    给你个例子,自己体会!

    # 0 1 1 2 3 5 8 13 ... ...
    
    def fib(n):
        a = 0
        b = 1
        count = 0
        while count < n:
            tep = a
            a = b
            b = tep + b
            # print(a, end=" ")
            yield a
            count += 1
    
    
    f = fib(20)
    print(next(f))
    print(next(f))
    print('-----"下面是上两个的和"-----')
    print(next(f))
    print(next(f))
    print(next(f))
    print(next(f))


这个东西还是比较抽象的,但是很重要的熬,所以建议多练习

自己试试输出一下斐波那契数列

好好看看!很重要的!!!!!!!!!!!!!!


*真的重要*
*必须康康*

猜你喜欢

转载自www.cnblogs.com/jevious/p/11142896.html