python判断水仙花数的代码怎么写

水仙花数是指一个 n 位数 (n ≥ 3)，它的每个位上的数字的 n 次幂之和等于它本身。例如，153 是一个水仙花数，因为 1^3 + 5^3 + 3^3 = 153。

以下是判断一个数是否为水仙花数的 Python 代码：

def is_armstrong_number(num):    # 将数字转化为字符串以获取位数    str_num = str(num)    # 获取数字的位数    n = len(str_num)    # 计算每个位上数字的 n 次幂之和    armstrong_sum = sum([int(digit)**n for digit in str_num])    # 判断是否为水仙花数    if armstrong_sum == num:        return True    else:        return False# 调用函数进行测试num = 153if is_armstrong_number(num):    print(num, "是水仙花数")else:    print(num, "不是水仙花数")

运行结果：

153 是水仙花数