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

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

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

在主函数中，可以输入一个整数，然后调用上述函数判断是否为素数，输出相应的信息：

n = int(input("请输入一个整数："))
if is_prime(n):
    print(n, "是素数")
else:
    print(n, "不是素数")