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

其中，`n**(0.5)` 表示 `n` 的平方根，因为一个数的因子不可能超过它的平方根。在这个代码中，我们从 2 开始遍历到 `n` 的平方根，如果发现 `n` 能被某个数整除，则 `n` 不是素数，返回 `False`，否则返回 `True`。