python用math判断一个数是素数

可以使用如下代码判断一个数n是否为素数：

import math

def is_prime(n):
    """判断一个数是否为素数"""
    if n < 2: # 小于2的数不是素数
        return False
    for i in range(2, int(math.sqrt(n))+1):
        if n % i == 0: # 如果n能被i整除，则不是素数
            return False
    return True

使用方法：

print(is_prime(5)) # True
print(is_prime(10)) # False

这里使用了sqrt函数来减少循环次数，提高效率。