生成一个可以判断素数的代码

以下是一个判断素数的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

该函数接受一个整数作为参数，返回一个布尔值。如果该整数是素数，则返回True，否则返回False。

该函数的实现思路是，从2开始，依次枚举小于等于n的所有整数，判断它们是否是n的因数。如果存在一个因数，则n不是素数，返回False；否则，n是素数，返回True。

需要注意的是，对于大于2的整数n，它的最小质因数肯定小于等于sqrt(n)，因此在枚举因数时只需要枚举到sqrt(n)即可。这样可以有效减少枚举量，提高判断素数的效率。