python写一判素数的函数

  • Post category:Python

下面是Python写一个判断素数的函数的完整攻略:

1. 确定函数名和参数

我们需要确定函数名和参数。对于判断素数的函数,我们可以起名为is_prime,它的参数为一个整数number。

2. 编写函数代码

根据素数的定义,只有 1 和它本身两个因数的数才是素数。因此,我们可以从2开始枚举,一直枚举到number-1,然后依次判断每个数能否整除number。若找到一个数能够整除number,则number不是素数;若枚举完毕,都没有找到能够整除number的数,则number是素数。

下面是 is_prime 的代码实现:

def is_prime(number):
    if number < 2:
        return False
    for i in range(2, number):
        if number % i == 0:
            return False
    return True

3. 代码说明

  • 首先判断传入的数是否小于2,若小于2,直接返回False。
  • 然后从2开始枚举到number-1,一直到找到能够整除number的数或者枚举完毕。
  • 在枚举过程中,如果找到能够整除number的数,则返回False,表示它不是素数。
  • 如果枚举完毕都没有找到能够整除number的数,则返回True,表示它是素数。

我们可以分别测试一下这个函数的输出,代码如下:

print(is_prime(2))    # True
print(is_prime(3))    # True
print(is_prime(4))    # False
print(is_prime(7))    # True

4. 代码示例说明

我们测试了is_prime函数对于一些数字的返回值,可以看到,它可以正确地判断2、3、4、7是否为素数,并给出相应的判断结果。

以上就是Python写一个判断素数的函数的完整攻略。