Python编写函数，判断一个数字是否为素数，是则返回字符串YES, 否则返回字符串NO,从键盘输入一个整数，调用该函数判断是否为素数。

以下是Python代码，用于判断一个数字是否为素数：

def is_prime(num):
    if num <= 1:
        return False
    for i in range(2, int(num ** 0.5) + 1):
        if num % i == 0:
            return False
    return True

n = int(input("请输入一个整数："))
if is_prime(n):
    print("YES")
else:
    print("NO")

你可以将上述代码复制到Python解释器中运行，输入一个整数，程序将输出YES或NO，表示该数字是否为素数。