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

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

def is_prime(n):
    if n < 2:
        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，那么它不是素数；如果一个数可以被2到它的平方根之间的任意一个数整除，那么它也不是素数；否则，它就是素数。