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

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

def is_prime(n):
    if n <= 1:
        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, "不是素数")

其中，`input()`函数用于从用户输入中获取一个字符串，`int()`函数将该字符串转换为整数。`if`语句判断该整数是否为素数，如果是，输出相应信息，否则输出另一种信息。