写一个对判素数的函数，在主函数输入一个整数，输出是否为素数的信息

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

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 = int(input("请输入一个整数："))
if is_prime(n):
    print(n, "是素数")
else:
    print(n, "不是素数")

这个函数的原理是，如果一个数不是素数，那么它一定可以分解成两个因数，其中至少一个小于等于它的平方根。因此，我们只需要在2到平方根之间枚举因数，如果存在一个因数能够整除这个数，那么它就不是素数。如果枚举完所有可能的因数都没有找到能够整除这个数的因数，那么它就是素数。