tomako123

导航

python迭代器和生成器

迭代器:

定义:

迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。

迭代器有两个基本的方法:iter() 和 next()。

字符串,列表或元组对象都可用于创建迭代器:

ex:

  #!/usr/bin/python3

  list=[1,2,3,4]
  it = iter(list)    # 创建迭代器对象
  for x in it:
      print (x, end=" ")
  输出:1 2 3 4

把类作为迭代器的方法:

要把类作为迭代器,得在类中实现两个方法 iter() 与 next()
iter() 方法返回一个特殊的迭代器对象, 这个迭代器对象实现了 next() 方法并通过 StopIteration 异常标识迭代的完成。

next() 方法(Python 2 里是 next())会返回下一个迭代器对象。

创建一个返回数字的迭代器,初始值为 1,逐步递增 1:

ex:

  class MyNumbers:
    def __iter__(self):
      self.a = 1
      return self

    def __next__(self):
      x = self.a
      self.a += 1
      return x

  myclass = MyNumbers()
  myiter = iter(myclass)

  print(next(myiter))
  print(next(myiter))
  print(next(myiter))
  print(next(myiter))
  print(next(myiter))

  输出结果为:
  1
  2
  3
  4
  5

stoplteration异常:

StopIteration 异常用于标识迭代的完成,防止出现无限循环的情况,在 next() 方法中我们可以设置在完成指定循环次数后触发 StopIteration 异常来结束迭代。
在迭代20次后停止执行:

		class MyNumbers:
		  def __iter__(self):
		    self.a = 1
		    return self
		 
		  def __next__(self):
		    if self.a <= 20:
		      x = self.a
		      self.a += 1
		      return x
		    else:
		      raise StopIteration
		 
		myclass = MyNumbers()
		myiter = iter(myclass)
		 
		for x in myiter:
		  print(x)
        结果是输出1到20

生成器:

定义:

使用了yield关键字的函数被称为生成器(generator)

yield 是一个关键字,用于定义生成器函数,生成器函数是一种特殊的函数,可以在迭代过程中逐步产生值,而不是一次性返回所有结果。

跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器。

当在生成器函数中使用 yield 语句时,函数的执行将会暂停,并将 yield 后面的表达式作为当前迭代的值返回。

然后,每次调用生成器的 next() 方法或使用 for 循环进行迭代时,函数会从上次暂停的地方继续执行,直到再次遇到 yield 语句。这样,生成器函数可以逐步产生值,而不需要一次性计算并返回所有结果。

调用一个生成器函数,返回的是一个迭代器对象。

  ex:
	def countdown(n):
	    while n > 0:
	        yield n
	        n -= 1
	 
	# 创建生成器对象
	generator = countdown(5)
	 
	# 通过迭代生成器获取值
	print(next(generator))  # 输出: 5
	print(next(generator))  # 输出: 4
	print(next(generator))  # 输出: 3
	 
	# 使用 for 循环迭代生成器
	for value in generator:
	    print(value)  # 输出: 2 1

posted on 2024-01-18 16:01  甜豆莎的辣白菜  阅读(9)  评论(0编辑  收藏  举报