生成器函数的使用(67)

1. 什么是生成器

• 生成器是迭代器的子类，可以把他当成一种特殊的迭代器，生成器支持迭代器所有的操作和特性

2. 生成器表达式

• a. 基本语法：tuple = ( i for i in xxx )

  set = (i for i in range(10))
  print(set)
  

• b. 推导式结合条件语句语法：tuple = ( i for i in xxx if 条件)
• c. 推导式结合三元运算符语法：tuple = (条件成立的值 if 条件 else 条件不成立的值 for i in xxx)

3. 什么是生成器函数

• 函数中使用了关键字 yield 的函数，都称之为生成器函数
• 生成器函数调用时不会直接执行，直接返回一个生成器对象（提示：

实例1：这是一个生成器函数：

	def work():
	    print("-------------------------- 1 --------------------------")
	    yield 1111
	    print("-------------------------- 2 --------------------------")
	    yield 2222
	    print("-------------------------- 3 --------------------------")
	    yield 3333
	    print("-------------------------- 4 --------------------------")
	    yield 4444
	
	res = work()
	print(res)    

• 去进行函数调用时，是没有返回任何数据的，哪怕 yield 返回了数据 2222

• 打印返回的是生成器对象：<generator object work at 0x000001C79DEAC930>
  

• 生成器函数必须使用生成器生成数据：next()函数，才会有数据出来，每次取出不重复的数据

  # 使用生成器生成数据：next()函数
  print("第一次生成的数据为：",next(res))
  print("第二次生成的数据为：",next(res))
  print("第三次生成的数据为：",next(res))
  print("第四次生成的数据为：",next(res))
  

  可以看到当前返回的数据：
  

• 当前这函数有四组数据，使用next()函数已经全部取出来了，第五次生成时已经没有数据了，会报错迭代器已经停止工作：StopIteration
  

4. 生成器的内部三个方法

• 生成器内部提供了三个方法：close方法，send方法，throw方法

1. close方法：手动关闭生成器

• 作用： 关闭生成器后，执行后续的操作，会报错迭代器已经停止工作：StopIteration

实例1：

	def work():
	    print("-------------------------- 1 --------------------------")
	    yield 1111
	    print("-------------------------- 2 --------------------------")
	    yield 2222
	    print("-------------------------- 3 --------------------------")
	    yield 3333
	    print("-------------------------- 4 --------------------------")
	    yield 4444
	
	
	# 创建一个生成器
	res = work()
	
	# 使用生成器生成数据：next()函数
	print("第一次生成的数据为：",next(res))
	
	#  使用close方法：手动关闭生成器，不在执行后面的步骤
	res.close()
	
	# 关闭生成器后，执行后续的操作，会报错迭代器已经停止工作：StopIteration
	print("第二次生成的数据为：",next(res))

执行结果：

• 可以看到在生成第一次的数据后，使用了close方法关闭生成器，报错迭代器已经停止工作：StopIteration
  

2. send方法：生成数据

• 和内置函数next一样，都是用来生成数据的
• 但是在生成数据的同时，可以和生成器内部进行数据交互

注意点：

• 使用send的方法之前，必须先通过next方法去生成至少一次数据（next去启动生成器），否则会报错
• 要使用send方法时，必须需要传一个参数（值）
• 在使用send() 方法传入时，会传到生成器内部上一次，挂起yield的哪里,用一个值接收，打印出来

实例2：

def work():
    print("-------------------------- 1 --------------------------")
    s1= yield 1111
    print("-------------------------- 2 --------------------------",s1)
    s2= yield 2222
    print("-------------------------- 3 --------------------------",s2)
    s3= yield 3333
    print("-------------------------- 4 --------------------------",s3)
    yield 4444

# 创建一个生成器
res = work()

print(res.send("hen1"))
print(res.send("hen2"))
print(res.send("hen3"))

执行结果会报错，为什么？

在使用send的方法之前，必须先通过next方法去生成至少一次数据才行

# 创建一个生成器
res = work()
print(next(res))  # 通过next方法生成一条数据

# 在使用send方法去传值
print(res.send("hen1"))
print(res.send("hen2"))
print(res.send("hen3"))

执行结果：

• 在使用send() 方法传入时，会传到生成器内部上一次，挂起yield的哪里
  

3. throw方法：主动在生成器内部抛出指定的异常

• 等同于在生成器内部执 raise ValueError，异常抛出

实例3：

def work():
    print("-------------------------- 1 --------------------------")
    s1= yield 1111
    print("-------------------------- 2 --------------------------",s1)
    s2= yield 2222
    print("-------------------------- 3 --------------------------",s2)
    s3= yield 3333
    print("-------------------------- 4 --------------------------",s3)
    yield 4444

# 创建一个生成器
res = work()

# 使用throw方法，抛出异常
res.throw(ValueError)   # 等同于在生成器内部执 raise ValueError

执行结果:
在这里插入图片描述