python定义一个判断质数的函数

  • Post category:Python

首先,我们需要了解什么是质数。质数是指在大于1的自然数中,只能被1和它本身整除的数,比如2、3、5、7等。

接下来,我们可以通过以下步骤定义一个判断质数的函数。

  1. 定义函数
def is_prime(number):

上述代码中,is_prime是定义的函数名称,number是该函数需要接收的输入参数。

  1. 特判1和小于1的整数
if number < 2:
    return False

如果输入的数字小于2,则不是质数,直接返回False。

  1. 判断可能的质数
for i in range(2, int(number**0.5)+1):
    if number % i == 0:
        return False
return True

从2开始,到该数的平方根范围内遍历一遍,如果存在能够被该数整除的整数,则该数不是质数,直接返回False。如果该数没有被整除,则表示该数是质数,返回True。

下面是完整的代码:

def is_prime(number):
    # 特判小于2的情况
    if number < 2:
        return False
    # 判断可能的质数
    for i in range(2, int(number**0.5)+1):
        if number % i == 0:
            return False
    return True

下面是一些示例代码的测试:

# 测试1:输入2,输出True
print(is_prime(2))

# 测试2:输入4,输出False
print(is_prime(4))

# 测试3:输入17,输出True
print(is_prime(17))

上面这段测试代码的运行结果如下:

True
False
True

可以看到,我们定义的函数可以有效地判断一个数字是否是质数。