python写一个判断素数的函数prim,在主函数输入一个整数,输出是否是素数的消息。

好的，我可以回答这个问题。以下是一个判断素数的函数prim的示例代码：

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

# 主函数
num = int(input("请输入一个整数："))
if prim(num):
    print(num, "是素数")
else:
    print(num, "不是素数")

希望能对你有所帮助！