编写函数,判断一个数字是否为素数,是则返回字符串YES,否则返回字符串NO。

好的，这是一个计算数学问题，我可以回答。以下是一个判断数字是否为素数的函数，你可以参考一下：

def is_prime(n):
    if n <= 1:
        return "NO"
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return "NO"
    return "YES"

你可以将需要判断的数字传入该函数，它会返回字符串"YES"或"NO"，表示输入的数字是否为素数。