欢迎访问宙启技术站
智能推送

如何在Python中编写函数来判断一个数是否为素数?

发布时间:2023-12-03 12:46:01

要判断一个数是否为素数,可以编写一个函数来实现。素数是指只能被 1 和自身整除的数。

下面是一个简单的函数来判断一个数是否为素数:

def is_prime(n):
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

这个函数的实现很简单。首先,如果 n 小于 2,则不是素数,直接返回 False。

接下来,我们通过一个循环从 2 开始,遍历到 n 的平方根(int(n ** 0.5) + 1)。如果在这个范围内找到了同时能被 n 整除的数,则 n 不是素数,返回 False。

如果循环结束后都没有找到能整除 n 的数,那么 n 是素数,返回 True。

我们可以用一些例子来测试这个函数:

print(is_prime(2))   # True
print(is_prime(3))   # True
print(is_prime(4))   # False
print(is_prime(29))  # True
print(is_prime(30))  # False

输出结果:

True
True
False
True
False

这些例子分别测试了一个质数、一个非质数(可以被 2 整除)、一个质数以及一个非质数(可以被 2 和 3 整除)。

通过这个函数,我们可以方便地判断一个数是否为素数。这对于解决一些与素数相关的问题非常有用,例如找出某个范围内的所有素数、判断一个数是否为完全平方数等。