用python写一个判断素数的函数

以下是用python编写判断素数的函数的代码：

def is_prime(n):
    """
    判断一个数是否为素数
    """
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

使用示例：

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

说明：

该函数接收一个整数n作为参数，如果n是素数，则返回True，否则返回False。在判断n是否为素数时，先判断n是否小于2，如果是，则直接返回False。然后从2开始遍历到n的平方根，如果n能够被任何一个数整除，则说明n不是素数，返回False。如果遍历了所有可能的因子，都没有找到能够整除n的数，则说明n是素数，返回True。