如何调用具有嵌套函数的函数？

Question

我是嵌套函数的新手，想在下面调用我的函数，我正在尝试使用嵌套函数和过滤器查找数字是否为素数。以下是我的代码：

def divides(n):
    def div(k):
        return n % k == 0
    return div

def prime(n):
    cnt=0
    res = list(filter(divides(n)(range(1,n)),range(1,n)))
    for i in res:
        if i==True:
            cnt+=1
    if cnt == 2:
        return ("It is a prime")
    else:
        return ("It is not a prime")

print(prime(5))

所以它会引发类型错误，因为我试图将数字从 1 到 n 作为范围发送到嵌套函数。如何使用两个值调用此函数？调用嵌套函数的正确方法是什么？

Answer 1

您的divides(n)(range(1,n)) 创建函数divides(n)，然后立即调用它，并使用全范围对象作为参数。该范围对象变为k，而n % k 失败。

您尝试使用filter 中的函数，它需要一个函数作为第一个参数：

res = list(filter(divides(n)(range(1,n)),range(1,n)))

你的意思是：

res = list(filter(divides(n), range(1, n)))

然后filter 将divides(n) 返回的函数应用于该范围内的每个数字。对于您的 n = 5，您将得到 res = [1]，这是该范围内 5 的唯一除数。

你用res 做的事情也有问题，看起来你应该使用range(1, n+1)，因为你试图找出是否正好有两个除数。只需检查 len(res) == 2 即可。

或者不是收集列表中的所有除数然后计数，您可以只检查是否有任何除数：

def prime(n):
    if any(filter(divides(n), range(2, n))):
        return ("It is not a prime")
    else:
        return ("It is a prime")

更简单，并且具有能够提前停止的额外好处。例如，prime(1000) 仅达到 2，不检查任何其他候选。