高阶函数之生成器和迭代器

迭代器和生成器

一、迭代器

1、迭代器的概念

• 迭代器：它是可迭代对象的一个子集，它是一个可以记住遍历的位置的对象。

2、迭代器的五条性质

---

• 1）打印迭代器无法查看其中的元素，也无法统计迭代器中的元素个数。

• 2）使用len方法也无法查看其中的元素。

• 3）迭代器类似数据结构中的队列（先进先出），迭代器中的元素被取出，即消失。

• 4）如果要获取迭代器中的元素，必须将元素从迭代器中一个一个的取出来。

• 5）任何容器型数据类型都可以通过 iter 方法转为迭代器。

---

 iter1 = iter('hello')
 iter2 = iter([1，2, 3])
 iter3 = iter((1, 2, 3))
 iter4 = iter({1, 2, 3})
 iter5 = iter({'name': '张三'，'age': 1})
 iter6 = iter(range(3))
 print(type(iter1))   
 print(type(iter2))    
 # 输出结果为：
 # <class 'str_iter
 # #<class 'list_iter 

3、获取迭代器中的元素

• 获取其中单个元素：next（每次只能取其中一个元素）

iter1 = iter('hello')
print(next(iter1))  # h
print(next(iter1))  # e
print(next(iter1))  # l
print(next(iter1))  # l
print(next(iter1))  # o
print(next(iter1))  # 报错：StopIteration

• 获取其中所有元素：for循环遍历。

 iter1 = iter('hello')
 for i in iter1:
     print(i)
 # 输出结果是竖行的：
 #   h
 #	e
 #	l
 #	l
 #	o

 print(next(iter1))   
 # 报错：StopIteration(迭代器元素取完消失)

4、迭代器的作用和意义

• a）在计算机内存中，元素被全部取出，迭代器消失，迭代器所占内存被释放，程序就可以减少内存占用。

二、生成器

1、生成器的概念

• a）生成器属于迭代器，但是迭代器不一定是生成器，元素取出即消失，和迭代器的性质类似；
• b）Python中使用了 yield关键字的函数 称为生成器。

2、生成器的性质

• a）在调用生成器函数时，函数体并不执行，只是返回一个生成器对象，只有在获取元素时生成器函数才执行，并且是获取一个元素才能够让生成器函数执行一次。

def func1():
	print('我是一个生成器函数')
	yield

result1 = func1()
# <generator object func1 at 0x7f7d780f1f90>
print(result1)  

• b）在对生成器对象使用next方法或者遍历的时候，生成器函数体才真正执行。

def func2():
	yield 100
	yield 200
	if 10 > 20:
   	yield 300
	return 400
	yield 500

result2 = func2()
'''
func2调用时只取一个元素,输出100，
for循环调用时，只剩200一个元素，取完元素消失
'''
for i in result2:
	print(i)
# 输出结果是：100    200

• c）生成器函数返回内部结果使用yield代替return。
• d）yield执行几次，生成器就产生几个元素。

3、生成器的意义及使用

---

• a）节省资源消耗，在不使用的时候几乎不占内存，也没有声明计算过程。

• b）使用的时候，生成器是随用随生成，用完即刻释放，非常高效。

• c）习题练习：写一个生成器可以创建Python学生的学号。

 def stuId():
 	for i in range(1, 1000):
 		yield f'python{i:0>3}'
 		
 f = stuId()
 print(f)
 # 从下边调用时才开始执行。
 for _ in range(1000):
 	print(next(f))          
 # 依次打印：python001 python002 ... 一直到python999