请用Python编写一个程序求解一个数是否为素数

以下是一个Python程序，用于检查一个数是否为素数。

def is_prime(n):
    # 0 和 1 不是素数
    if n < 2:
        return False
    # 2 是素数
    elif n == 2:
        return True
    # 偶数不是素数
    elif n % 2 == 0:
        return False
    # 检查奇数是否为素数
    else:
        for i in range(3, int(n ** 0.5) + 1, 2):
            if n % i == 0:
                return False
        return True

这个程序首先检查输入的数是否小于2，如果是，则返回False，因为0和1不是素数。如果输入的数是2，则返回True，因为2是素数。如果输入的数是偶数，则返回False，因为除2以外的偶数不是素数。最后，程序通过检查从3到n的平方根之间的奇数来判断奇数是否为素数。如果n可以被任何从3到n的平方根之间的奇数整除，则n不是素数，否则n是素数。