定义一个函数is_prime（n)判断输入的n是否为素数

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

该函数首先判断输入的n是否小于等于1，因为小于等于1的整数都不是素数。然后从2开始遍历到n的平方根（加1取整），如果n能被其中的一个数整除，则n不是素数，返回False。如果遍历完整个范围都没有找到能整除n的数，则n是素数，返回True。