Python，编写程序判断素数

以下是 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

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