判斷一個整數是否為素數

"""
    輸入一個整數N，判斷這個整數是否是素數（prime）
     注：素數是只能被1和自身整除的數。
     方法：
        用排除法，一旦n能被2~n-1的數整除就不是素數，否則就一定是素數
"""
#方法一：
integer = int(input('請輸入一個整數：'))
if integer < 2:
    print(integer,'不是素數')
    exit()
flag = True #true代表是素數，false代表不是素數
for ax in range(2,integer):
    if integer % ax ==0:
        flag = False
        break
if flag == True:
    print(integer,'是素數！')
else:
    print(integer,'不是素數！')


#方法二：
int_n = int(input('請輸入一個整數：'))
if int_n < 2:
    print(int_n,'不是素數')
    exit()
for bx in range(2,int_n):
    if bx % 2 ==0:
        print(int_n,'不是素數！')
else:
    print(int_n,'是素數！')