编写Python函数，判断一个数字是否为素数，是则返回字符串 YES，否则返回字符

好的，下面是代码实现：

def is_prime(n):
    if n < 2:  # 小于2的数不是素数
        return "NO"
    for i in range(2, int(n**0.5) + 1):  # 判断是否有因子
        if n % i == 0:
            return "NO"
    return "YES"

使用方法：

print(is_prime(17))  # YES
print(is_prime(21))  # NO

注意：该函数只能判断正整数是否为素数。