Python利用生成器实现可迭代对象的方法

向北走

发布时间：2026-10-13| 4人浏览过

原创

本文介绍如何在Python中通过生成器实现自定义可迭代对象。以输出指定范围内素数为例，演示了如何重写__iter__方法，利用yield关键字逐个返回素数，从而支持for循环遍历。

本文分享了一个使用Python生成器来构建可迭代对象的具体代码示例，供读者参考学习。

具体案例分析如下：

我们需要实现一个自定义的类，使其能够迭代输出给定区间内的所有素数。

实例化类：pn = Number(1, 30)

遍历该对象：for k in pn:

打印当前迭代的值：print(k)

最终输出的结果为：2, 3, 5, 7, 11, 13, 17, 19, 23, 29

针对上述需求，具体的解决思路是什么？

核心在于将该类的__iter__方法实现为生成器函数。在迭代过程中，每次遇到素数时，使用yield语句返回该素数。

#!/usr/bin/python3
 
 
class Number(object):
 def __init__(self, start, end):
  self.start = start
  self.end = end
   
 # 判断一个数字是否是素数
 def get_num(self, k):
  if k >= 2:
   for i in range(2, k):
    if k % i == 0:
     return False
   return True
  
 def __iter__(self):
  for k in range(self.start, self.end+1):
   if self.get_num(k):
    # 是素数yield出去
    yield k
     
if __name__ == '__main__':
 num = Number(2, 30)
 for i in num:
  print(i)

以上就是本文的全部内容，希望能为读者的学习提供有益的帮助，同时也欢迎大家继续支持本站。