素數（數據結構與算法）（python）

編寫一個函數Prime（n），對於已知正整數n，判斷該數是否爲素數，如果爲素數，返回True，否則返回False。（python）

def Prime(n):
    if n<2:
        return False
    elif n==2:
        return True
    else:
        i=2
        for i in range(2,n):
            if n%i==0:
                break
            else:
                i=i+1                
        if i==n:
            return True
        else:
            return False

n=int(input("請輸入一個數："))
print(Prime(n))