判断输入数据是质数还是合数

# 判断一个数是否为质数

num = int(input("请输入需要判断的数字"))
is_prime = True
if num < 1:
    print("输入的数据不合法")
elif num == 1:
    print("1 既不是质数也不是合数")
elif num == 2:
    print("2是质数")
else:
    for i in range(2,num):
        if num % i == 0:
            is_prime  =  False
            break
    if(is_prime == True):
        print("{}是质数".format(num))
    else:
        print("{}是合数".format(num))