输入一个数，判断是否为素数

以下是判断素数的 Python 代码：

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

你可以调用这个函数来判断一个数是否为素数。