python中用while和if判斷質數

#從控制台輸入一個數，判斷是否為質數
a=int(input("輸入一個數："))
b=2
e=0
while b<=a:
    c=a%b
    if c==0:
        e += 1
    b+=1
if e>1:
    print("不是質數")
else:
    print("是質數")

前一個輸入1會出錯

---

#從控制台輸入一個數，判斷是否為質數
a=int(input("輸入一個數："))
b=1
e=0
while b<=a :
    c=a%b
    if c==0:
        e += 1
    b+=1
if e==2:
    print("是質數")
else:
    print("不是質數")