Python3-判断一个数是否输出素数

#判断输入的数是否是素数
#2021年6月14日

def sushu(num):
    if num>1:
        for i in range(2,num):
            if num%i==0:
                print(num,"不是素数！",i,"*",num//i,"等于",num)
                break
        else:
            print(num,"是素数！")
    else:
        print(num,"小于1，不是素数")

if __name__ == "__main__":
    n=int(input("请输入一个数字："))
    sushu(n)