用代码编写函数判断一个数是否为素数

可以使用如下的Python代码编写函数来判断一个数是否为素数：

def is_prime(n):
    if n <= 1:
        return False
    elif n <= 3:
        return True
    elif n % 2 == 0 or n % 3 == 0:
        return False
    i = 5
    while i * i <= n:
        if n % i == 0 or n % (i + 2) == 0:
            return False
        i += 6
    return True

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

该函数的实现基于著名的素数判定算法——Miller-Rabin 算法，该算法的时间复杂度为 $O(k\log^3n)$，其中 $k$ 是算法的迭代次数，通常取 $k=20$ 就足够了。