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

其中，`int(n ** 0.5) + 1` 表示对 n 开根号向下取整再加 1，可以提高算法的效率。