编写一个函数判断素数

以下是 Python 代码实现：

def is_prime(n):
    """
    判断一个数是否为素数
    """
    if n < 2:  # 小于2的数不是素数
        return False
    for i in range(2, int(n**0.5)+1):  # 只需遍历到 n 的平方根
        if n % i == 0:
            return False
    return True

该函数接受一个数 `n` 作为参数，如果 `n` 是素数则返回 `True`，否则返回 `False`。

该函数的实现思路是，先判断 `n` 是否小于 2，若小于 2 则直接返回 `False`，因为小于 2 的数都不是素数。接着使用一个 for 循环遍历 2 到 `n` 的平方根之间的所有整数，判断是否能够被 `n` 整除，若能则说明 `n` 不是素数，返回 `False`。如果整个循环都没有返回 `False`，则说明 `n` 是素数，返回 `True`。