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

以下是一个判断素数的函数的示例代码：

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, "不是素数")

函数 `is_prime(n)` 的作用是判断整数 `n` 是否为素数，如果是素数则返回 `True`，否则返回 `False`。该函数的实现方法是：从 2 开始到 $\sqrt{n}$（向下取整）的范围内依次判断 `n` 是否能被整除，如果能则说明 `n` 不是素数，返回 `False`；如果没有找到能整除 `n` 的数，则说明 `n` 是素数，返回 `True`。

在主函数中，首先输入一个整数 `n`，然后调用 `is_prime(n)` 函数判断 `n` 是否为素数，最后根据返回值输出相应的信息。