Python利用生成器实现可迭代对象的方法
发布时间：2026-10-13 |
1人浏览过
本文介绍如何在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)
以上就是本文的全部内容，希望能为读者的学习提供有益的帮助，同时也欢迎大家继续支持本站。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
