Python 快速判斷一個數是不是素數

n = int(input('Input an integer: '))
if n == 2:
    print('Yes')
elif n % 2 == 0:
    print('No')
else:
    m = n % 6
    #因為大於5的素數必然出現在6的倍數兩側
    #6x+2,6x+3,6x+4肯定不是素數
    if m != 1 and m != 5:
        print('No')
    else:
        for i in range(3, int(n**0.5) + 1, 2):
            if n % i == 0:
                print('No')
                break
        else:
                print('Yes')