编一个函数判断一个数是否为素数

以下是判断一个数是否为素数的Python函数：

def is_prime(num):
    if num < 2:
        return False
    for i in range(2, int(num ** 0.5) + 1):
        if num % i == 0:
            return False
    return True

该函数接受一个整数作为参数，如果该整数小于2，则返回False；否则，从2到该整数的平方根遍历每个数，如果该整数能被遍历到的数整除，则返回False，否则返回True。