确定随机 Int 是否为素数

Question

当我需要确定一个数字 x 是否是素数时，我遇到了一个小问题。 x 将是一个随机生成的正整数，我在执行代码时收到以下消息：

Your function fails on is_prime(2). It returns None when it should return True.

我的代码：

def is_prime(x):
if x < 2:
    return False
else:
    for n in range(2, x):
        while x % n == 0:
            return False
            break
        else:
            return True

我希望 while 循环遍历 n == 2 直到 n == (x-1) 但它似乎没有这样做！ 我做错了什么？

Answer 1

你可以更简单地编写这个函数：

import math
def is_prime(n):
    if n % 2 == 0 and n > 2: 
       return False
    for i in range(3, int(math.sqrt(n)) + 1, 2):
        if n % i == 0:
           return False
    return True

因此，您对 is_prime(2) 的调用不会通过 if 测试，也不会在 for 运行中使用，只会返回 True。

如果其他素数大于 3 且不是偶数，则应测试它们。