编写函数 判断一个数字是否为素数 是返回True

下面是 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

函数中，我们首先判断如果数字小于 2，那么它不是素数，直接返回 False。然后我们用一个 for 循环，从 2 开始到 $\sqrt{n}$（向下取整）为止，如果这个数字能被任何一个小于它的数字整除，那么它就不是素数，返回 False。如果循环结束后都没有找到能整除它的数字，那么它就是素数，返回 True。