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

以下是Python编写函数，判断一个数字是否为素数的代码：

def is_prime(num):
    if num < 2:
        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解释器中运行，输入一个整数，程序会判断该数是否为素数并输出结果。