用python写一个判断素数

以下是一个用 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

这个程序的思路是：如果一个数小于 2，则不是素数；如果一个数大于等于 2，则从 2 开始到它的平方根，依次判断是否能整除，如果能整除，则不是素数，否则是素数。

可以通过调用这个函数来判断一个数是否是素数，例如：

print(is_prime(17))  # True
print(is_prime(18))  # False

这个程序的时间复杂度为 $O(\sqrt{n})$，其中 $n$ 是待判断的数。