python3判斷一個數是否為素數

while True:
    num = int(input('請輸入一個數：'))
    for i in range(2,num):#判斷在num之前的數能不能把num整除
        if(num%i == 0):
            print('%d不為素數'%num)
            break
        else:
            print('%d為素數'%num)
            break