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

  • Post category:Python

首先,定义一个判断质数的函数需要判断输入的数字是否是质数。质数是指只有1和本身两个因数的数,因此需要从2开始,一直判断到该数的平方根,是否存在其他可以整除该数的因数。如果存在,则该数不是质数,否则是质数。

以下是完整的python代码实现:

import math

def is_prime(num):
    if num <= 1:
        return False

    # 判断是否是质数
    for i in range(2, int(math.sqrt(num))+1):
        if num % i == 0:
            return False
    return True

上述代码中,import math导入math库,可以使用其中的sqrt函数计算数字的平方根。

def is_prime(num):定义了一个名为is_prime的函数,函数里面传入一个参数num。

if num <= 1:判断输入的数字是否小于等于1,如果小于等于1则不是质数,返回False。

for i in range(2, int(math.sqrt(num))+1):使用for循环从2到该数字的平方根进行遍历,如果该数字可以被整除,则不是质数,返回False。

return True如果没有返回False,则该数字是质数,返回True。

下面是两个实例,来看看函数的使用:

print(is_prime(7))  # True
print(is_prime(15)) # False

上述两个实例分别传入数字7和15,第一个数字是质数,第二个数字不是质数,函数正确地给出了判断结果。