编写一个判断素数的函数

以下是一个用于判断素数的函数实现，它接收一个整数作为参数，如果该整数是素数，则返回True，否则返回False。

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

该函数首先判断输入的整数是否小于2，因为小于2的整数都不是素数。接着，从2到n的平方根之间的每个整数i进行迭代。如果n可以被i整除，则n不是素数，并且函数返回False。如果循环结束后没有找到任何可以整除n的数，则n是素数，函数返回True。