python3基础:生成器

2023-01-31 01:01:26 基础 生成器 python3

python中,边循环边计算的机制就称为生成器(generator)。使用生成器当用到数据的时候在生成,这样可以节约空间,提高效率。生成器是一种特殊类型的迭代器,所以也会有__init__()和__next__()两个方法:

__iter__方法负责返回一个迭代器(在迭代器中返回自己,在可迭代对象中返回帮助自己迭代的迭代器)

_next__方法:当前要获取的元素没有越界,就返回当前元素,然后自己指向为下一个元素等待返回;如果上次反回了最后一个元素,这一次再调用next的时候已经没有元素了,就抛出StopIteration异常。

1.( ) 括号内 放入列表推倒表达式 返回一个生成器对象

  • 用next()方法取出元素:

myIterator = ( x*2 for x in range(5))
print(type(myIterator))

print(next(myIterator))
print(next(myIterator))
print(next(myIterator))
print(next(myIterator))
print(next(myIterator))
print(next(myIterator))#报错,因为没有元素可以取了

执行结果:

 

说明:

 写法和推导列表非常相似,只是由[]换成了()。由于生成器也是迭代器,所以可以使用next()方法将元素一个一个取出来,当所有的元素都取出后再调用next()方法就会抛出StopIteration异常。

  • 利用for循环取出元素:

myIterator = ( x*2 for x in range(11))
print(type(myIterator))
for i in myIterator:
    print(i,end=" ")

执行结果: 

 

说明:

由于生成器也是迭代器所以可以放在for循环中操作,将元素一个一个取出来。可以发现使用这种方法不会有异常抛出,因为for循环内部已经处理了。

2.yield关键字

比如我们定义一个函数,打印生成n以内的整数,代码如下:


def my_print(n):
      for i in range(n):
            print(i)

my_print(5)

 执行结果:

当把print换成yield之后,这个函数就是一个生成器了,代码如下:


def my_print(n):
      for i in range(n):
            yield(i)

print(my_print(5))
for i in my_print(5):
      print(i)

执行结果:

 

说明:

只是由print改成了yield就得到了一个生成器对象,可以用next()或者for循环将数据取出来。换成yield之后到底发生了什么?

当函数执行到yield i 的时候,函数会把i的数值抛出来,我们用for循环遍历的时候获取了yield 后面的值,然后函数就会暂停,等待下一次遍历的时候,函数从yield继续向下执行,直到遇到yield的时候又返回了i的值,然后函数再暂停,等待下一次唤醒。这个循环一直做,到函数结束的时候。

1.用生成器生成杨辉三角


def triangles(n):
      temp_list= [1]
      for i in range(1,n+1):
            yield temp_list
            #每一行是一个list,由上一次计算的结果中每相邻的两个元素之后作为新列表的中间值
            temp_list = [temp_list[i]+temp_list[i+1] for i in range(len(temp_list)-1)]
            #第一个位置中添加1
            temp_list.insert(0,1)
            #最后一个位置中添加1
            temp_list.append(1)

for i in triangles(10):
      print(i)

 

2.以下代码的执行结果是什么?


def add(n, i):
      return n+i

def test():
      for i in range(4):
            yield i

g = test()
for n in [1,10,5]:
      g = (add(n,i) for i in g)

print(list(g))

分析:

g只是代表一个生成器对象,没有for,next,list,tuple调用它的时候它是不会执行的,真实的代码执行逻辑如下:

第1次:n = 1, 执行g=(add(n,i) for i in test())   
第2次:n = 10,执行g=(add(n,i) for i in (add(n,i) for i in test())) )  
第3次:n = 5,执行g=(add(5,i) for i in (add(n,i) for i in (add(n,i) for i in test())) ))
到第3次的时候n的值就是5,也就是说前两次给n赋的值会覆盖掉,或者可以这样理解,上面的代码等价于:


def add(a,b):            
    return a + b
def test():              
    for r in range(4):
        yield r
g=test()
n=2
g=(add(n,i) for i in g)
n=10  #覆盖前面n的值
g=(add(n,i) for i in g)
n=5  #覆盖前面n的值
g=(add(n,i) for i in g)
print(list(g))

 g因为前面做过定义,每次需要把之前定义的带入,所以当n=5 的时候,g=(add(5,i) for i in (add(n,i) for i in (add(n,i) for i in test())) )),最终的执行结果是[15, 16, 17, 18]

1.碰到for循环套生成器,就拆开来做 ;

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

 

参考链接:

https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/0014317799226173f45ce40636141b6abc8424e12b5fb27000

Https://www.cnblogs.com/gkx0731/p/9538348.html

 

相关文章