python定义函数求素数

函数定义

在Python中，我们可以使用def关键字来定义一个函数，素数是一个大于1的自然数，除了1和它本身以外不再有其他因数，我们可以通过检查一个数是否有其他因数来判断它是否是素数。

函数解析

1、我们需要定义一个函数，名为is_prime，它接受一个参数n。

2、我们需要判断n是否小于等于1，如果是，那么它就不是素数，直接返回False。

3、接下来，我们需要遍历从2到n的平方根的所有整数，如果n可以被其中任何一个整数整除，那么它就不是素数，直接返回False。

4、如果n不能被2到n的平方根的任何整数整除，那么它就是素数，返回True。

代码实现

import math
def is_prime(n):
    # 判断n是否小于等于1
    if n <= 1:
        return False
    # 遍历从2到n的平方根的所有整数
    for i in range(2, int(math.sqrt(n)) + 1):
        # 如果n可以被i整除，那么它就不是素数
        if n % i == 0:
            return False
    # 如果n不能被2到n的平方根的任何整数整除，那么它就是素数
    return True

单元测试

我们可以使用一些测试用例来验证我们的函数是否正确，2、3、5、7、11、13、17、19、23都是素数，而4、6、8、9、10、12、14、15、16都不是素数。

print(is_prime(2))  # True
print(is_prime(3))  # True
print(is_prime(4))  # False
print(is_prime(5))  # True
print(is_prime(6))  # False
print(is_prime(7))  # True
print(is_prime(8))  # False
print(is_prime(9))  # False
print(is_prime(10)) # False
print(is_prime(11)) # True
print(is_prime(12)) # False
print(is_prime(13)) # True
print(is_prime(14)) # False
print(is_prime(15)) # False
print(is_prime(16)) # False
print(is_prime(17)) # True
print(is_prime(18)) # False
print(is_prime(19)) # True
print(is_prime(20)) # False
print(is_prime(23)) # True