判断素数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 作为输入，如果 n 是素数则返回 True，否则返回 False。算法的思路是从 2 开始一直到 n 的平方根进行循环，如果在这个范围内找到了一个数能够整除 n，则 n 不是素数，返回 False。如果循环结束还没有找到可以整除 n 的数，则 n 是素数，返回 True。