时间戳

首页 > 解决方案 > Pyhon 生成器函数中的异常处理

python - Pyhon 生成器函数中的异常处理

问题描述

我必须创建 Python 函数生成器，它应该返回正数的所有除数。如果没有除数，left 函数应该返回 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？我究竟做错了什么？如何解决？

标签： pythonexceptiongenerator

解决方案

发生这种情况是因为该函数不是生成器。它是一个常规函数，final在第一次调用时从变量返回生成器。在这种情况下，最好的解决方案是将函数转换为实际的生成器并完全省略错误处理：

def divisor(num):
    for x in range(1, num + 1):
        if num / x == int(num / x):
            yield x
    while True:
        yield None

推荐阅读