如何在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 整除)。
通过这个函数,我们可以方便地判断一个数是否为素数。这对于解决一些与素数相关的问题非常有用,例如找出某个范围内的所有素数、判断一个数是否为完全平方数等。
