【python】7-6 输入一个数，判断是不是水仙花数

输入一个整数，判断它是不是水仙花数，水仙花数是一个三位整数，其每位数字的立方和等于其本身，
例如，153是一个水仙花数，153=1×1×1+5×5×5+3×3×3

num=int(input())
ge=num%10
shi=num//10%10
bai=num//100
if num <100 or num >999:
    print("输入错误，不是一个三位数")
elif ge**3+shi**3+bai**3==num:
    print("该数是水仙花数")
else:
    print("该数不是水仙花数")