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

好的，以下是一个判断素数的函数 `is_prime()` 的实现：

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`，函数会返回一个布尔值，表示 `n` 是否为素数。如果 `n` 是素数，返回 `True`，否则返回 `False`。

在主函数中，你可以通过以下代码调用 `is_prime()` 函数：

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

这里的 `input()` 函数用于从用户输入中获取一个整数，然后判断它是否为素数。