主要是依據質數符合6k+-1的規律，而取除數范圍是2至根號n

            
              import math




def find_primery(number):
    #判斷1，2，3
    if number in [2,3]:
        return 1
    elif number == 1:
        return 0
    #能被2或3整除
    if number%2 == 0 or number%3 == 0:
        return 0#合數
    #質數判別6k+-1
    elif (number+1)%6 == 0 or (number-1)%6==0:
        pass
    else:
        return 0 #合數
    
    squrt_n = int(math.sqrt(number))#計算上限
    if int(squrt_n+1)/6 > 1:#確定步進上限
        step_max = int((squrt_n+1)/6)
    else :
        step_max = 1
    
    for loop in range(1,step_max):#1到step_max為在2至根號number之間的質數的步進范圍
        print(loop)
        if number%(loop*6+1)==0 or number%(loop*6-1)==0:#除以此范圍內的所有質數
            return 0
        else:
            pass #合數
        
    return 1 #他是質數
    

if __name__ == '__main__':
    while 1:
        str1 = input('input number:')
        number = int(str1)
        print('result:%d'%(find_primery(number)))
            
          

?