（python 3）判断素数

python 3判断素数函数

代码： 

#函数用来判断是否为素数
def isprime(n):
    #n为2或3，可以直接判断是素数
    if n == 2 or n == 3:
        return True
    #n可以被2或3整除，可以直接判断不是素数
    if n % 2 == 0 or n % 3 == 0:
        return False

    #观察素数的分布可以发现，除2&3以外的素数，必定分布在6k（k为大于1的整数）的两侧。
    #6k%6==0
    #(6k+2)%2==0
    #(6k+3)%3==0
    #(6k+4)%2==0
    #所以2&3以外的素数为6k+1或6k-1的形式，据此可以缩小因子范围
    for k in range(6, int(math.sqrt(n)) + 2, 6):
        if n % (k - 1) == 0 or n % (k + 1) == 0:
            return False
    return True