在Python中,你可以编写一个函数来判断一个数是否为素数。素数是指只有1和它本身两个正因数(不包括1本身)的自然数。以下是一个简单的示例代码:
python复制代码
def is_prime(n): | |
if n <= 1: | |
return False | |
if n <= 3: | |
return True | |
if n % 2 == 0 or n % 3 == 0: | |
return False | |
i = 5 | |
while i * i <= n: | |
if n % i == 0 or n % (i + 2) == 0: | |
return False | |
i += 6 | |
return True | |
# 测试代码 | |
num = int(input("请输入一个整数: ")) | |
if is_prime(num): | |
print(num, "是素数") | |
else: | |
print(num, "不是素数") |
在这个代码中,is_prime
函数用于判断一个数是否为素数。首先,我们检查一些基本情况:如果 n
小于等于 1,那么它肯定不是素数;如果 n
小于等于 3,那么它是素数(因为 2 和 3 都是素数)。然后,我们检查 n
是否可以被 2 或 3 整除,如果可以,则 n
不是素数。接下来,我们使用一个循环来检查 n
是否可以被其他数整除。这个循环从 5 开始,每次增加 6(因为我们已经检查了所有形如 6k ± 1 的数)。如果 n
可以被循环中的任何数整除,那么 n
不是素数。如果循环结束而没有找到任何能整除 n
的数,那么 n
是素数。
最后,我们使用 input
函数从用户那里获取一个整数,并使用 is_prime
函数来判断这个数是否为素数,然后输出结果。