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

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

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

if __name__ == "__main__":
    main()

上面是一个简单的 Python 素数判断函数。在主函数中，用户输入一个整数并调用 is_prime() 函数判断是否为素数，最后输出结果。