在python中判斷一個數是否為水仙花數

153 = 1³+ 5³ + 3³-1 + 125 +27 =153

#(1)輸入一個3位數n
n = input("輸入一個三位數")

##將字符串 轉成 數值 判斷 n  是否為3位數，
n = int(n)
if n > 999 or n < 100:
    print("請輸入三位數")
    exit(1)

#(2)獲取各個位數  123
a = n//100
b = n%100//10
c = n%10
#(3)計算各個位數 立方 之和sum
sum = a**3 + b**3 + c**3
#(4)判斷立方和sum   和 數值n是否相等
if sum == n:
    print("是水仙花數")
else:
    print("不是水仙花數")