創建用戶定義的 function 以檢查 python 中的素數

Question

num=int(input("enter the no: "))
def Prime(num):

    """Check whether a no is prime or not"""
    for i in range(2,num):

        if num%i==0:
            print(num," is not prime no")
            break
        else:
            print(num,"is prime no")
            break

print(Prime(num))

而 output 是這樣的

    enter the no: 45
    45 is prime no
    None

這段檢查素數的代碼有什么問題，以及為什么在 output 中打印“None”。我如何在此處使用 return。 output 對於一些不正確的 97 是正確的，但在這里對於 45 是錯誤的。非常感謝您的建議。

Answer 1

num=int(input("enter the no: "))

def Prime(num):
    flag = False
    if num > 1:
        for i in range(2, num):
            if (num % i) == 0:
                flag = True
                break

    if flag:
        print(num, "is not a prime number")
    else:
        print(num, "is a prime number")
        
Prime(num)