python基础——yield关键字、生成器、迭代器

python的yield是实现生成器的关键字，yield关键字可以记住当前程序执行的位置，并在下一次调用时从yield处开始执行

a = ["aaa", "bbb", "ccc", "ddd"]

def t():
    print(f"t function start")
    for index, value in enumerate(a, start=1):
        print(f"index is {index}, value is {value}")
        yield index, value
        print(f"xunhuan {index}")

def w():
    for i in t():
        print(f"w function {i}")
        yield "xiaogu"
        print(f"w function end")

for i in w():
    print(i)

以以上代码为例：首先for循环调用w()时，去调用t()函数，在t函数中，执行到yield时，程序返回index和value，返回到程序的调用处，即w函数中的for循环，程序继续向下执行，在执行到yield “xiaogu”时，程序接着返回，返回到for循环调用w函数处，返回值赋值给i，打印。接下来程序继续for循环调用w函数，此时程序会先打印“w function end”，再执行for循环调用t函数，先从程序暂停的地方开始执行，即先打印"xunhuan 1"，再执行for循环，以后执行顺序和刚开始相同，直到最后一个for循环程序退出，即t函数中的for循环执行完毕，程序自动退出。

生成器是一种特殊的迭代器，通过yield关键字自动实现了next方法，不需要手动执行

迭代器的实现需要一个对象中实现__iter__方法和__next__方法 ，迭代器是帮我们记住每次访问的位置，在执行next函数的时候迭代器会返回记住位置的下一位置数据

可迭代对象只要实现__iter__方法即可称为一个可迭代对象，可迭代对象中如果没有实现自己的迭代器是不可以进行迭代的

举例：
如果一个函数中有yield关键字，那这个函数就不是一个函数而是一个生成器
如果一个类中的某个方法中含有yield关键字，也是可以实现生成器进行迭代

def func(num):
	while num < 10:
		num += 1
		yield num 

for i in func(0):
	print(i)
# 输出：1 2 3 4 5 6 7 8 9 10

class A:
	def add(self, num):
		while num < 10:
			num += 1
			yield num

for i in A().add(0):
	print(i)
# 输出：1 2 3 4 5 6 7 8 9 10

实现一个迭代器， 同时实现iter方法和next方法

class A:
	def __init__(self, num):
		self.num = num
		self.current = 0

	def __next__(self):
		while True:
			if self.current < self.num:
				self.current += 1
				return self.current  
			else:
				raise StopIteration
				
	def __iter__(self): 
		return self

for i in A(10):
	print(i)
# 输出：1 2 3 4 5 6 7 8 9 10

实现一个可迭代对象

from collections import Iterable

class A:
	def __iter__(self):
		pass

print(isinstance(A(), Iterable))
# 输出：True

# 这个可迭代对象是不可以进行迭代的，因为他没有实现自己的迭代器
# 而for循环的本质就是去调用可迭代对象的迭代器实现迭代
for i in A():
	print(i)
# 输出：TypeError: iter() returned non-iterator of type 'NoneType'