如何解决Pyhon生成器函数中的异常处理
我必须创建Python函数生成器,该函数应返回正数的所有除数。 如果没有除数,则函数应返回None。 这样完成:
def divisor(num):
try:
final = (x for x in range(1,num + 1) if num / x == int(num / x))
except StopIteration:
return None
else:
return final
获得这样的结果:
three = divisor(3)
next(three) => 1
next(three) => 3
但是当我再次调用该函数 时,出现 StopIteration 错误:
next(three) =>
Traceback (most recent call last):
...
StopIteration
为什么不处理StopIteration?我究竟做错了什么?如何解决?
解决方法
之所以发生,是因为该函数不是生成器。它是一个常规函数,它在第一次调用时从final
变量返回生成器。在这种情况下,最好的解决方案是将函数转换为实际的生成器并完全忽略错误处理:
def divisor(num):
for x in range(1,num + 1):
if num / x == int(num / x):
yield x
while True:
yield None
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。