python，自定义函数，用户输入一个数，判断这个数是否是素数，如果是打印yes，如不不是打印no

我可以回答这个问题。以下是一个用 Python 编写的函数，用于判断一个数是否是素数：

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

这个函数接受一个参数 n，然后通过循环来检查 n 是否是素数。如果 n 小于 2，直接返回 False。否则，从 2 开始到 n 的平方根（向上取整）的所有数中循环，检查是否有能整除 n 的数。如果有，说明 n 不是素数，返回 False。如果循环结束后都没有找到能整除 n 的数，说明 n 是素数，返回 True。如果要使用这个函数来判断用户输入的数是否是素数，可以将用户输入的数作为参数传递给 is_prime 函数，然后根据返回值打印 Yes 或者 No。