迭代器
迭代器是一个可以记住遍历的位置的对象。
迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
迭代器有两个基本的方法:iter() 和 next(),可以通过next()和for循环进行调用
字符串,列表或元组对象都可用于创建迭代器:
it2 = iter("abcde")
print(next(it2))
for x in it2:
print(x, end=' ')
########### output
a
b c d e
lis = [1,2,3,4]
it = iter(lis)
for x in it:
print(x, end=' ')
########## output
1 2 3 4
构造一个迭代器
class MyNumbers:
def __iter__(self):
self.a = 1
return self
def __next__(self):
if self.a < 10:
x = self.a
self.a += 1
return x
else:
raise StopIteration
myclass = MyNumbers()
myiter = iter(myclass)
print(next(myiter))
print(next(myiter))
############ output
1
2
生成器
使用了 yield 的函数被称为生成器(generator),yield 是一个关键字,用于定义生成器函数,生成器函数是一种特殊的函数,可以在迭代过程中逐步产生值,而不是一次性返回所有结果
调用一个生成器函数,返回的是一个迭代器对象。
当在生成器函数中使用 yield 语句时,函数的执行将会暂停,并将 yield 后面的表达式作为当前迭代的值返回。
然后,每次调用生成器的 next() 方法或使用 for 循环进行迭代时,函数会从上次暂停的地方继续执行,直到再次遇到 yield 语句。这样,生成器函数可以逐步产生值,而不需要一次性计算并返回所有结果。
def countdown(n):
while n > 0:
yield n#每次调用yeild关键字时,函数返回当前的值,下一次调用时从当前位置开始,直到遇到下一个yield
n -= 1
# 创建生成器对象
generator = countdown(5)
# 通过迭代生成器获取值
print(next(generator)) # 输出: 5
print(next(generator)) # 输出: 4
# 使用 for 循环迭代生成器
for value in generator:
print(value,end=' ') # 输出: 2 1
############### output
5
4
3 2 1