编写一个函数判断素数

下面是 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是否为素数。如果n小于2，函数直接返回False。否则，函数遍历从2到n的平方根的所有整数，检查它们是否能够整除n，如果能够整除，则n不是素数，函数返回False。如果遍历结束后都没有找到能够整除n的整数，说明n是素数，函数返回True。