用python写水仙花代码

在Python中编写检查数字是否是“水仙花数”(又称“ narcissistic number”或“自恋数”)的代码非常简单。水仙花数是指那些每一位数字的立方和等于其本身的三位数。例如，153是一个水仙花数，因为1^3 + 5^3 + 3^3 = 153。

下面是一个简单的函数来检查输入的数字是否是水仙花数：

def is_narcissistic_number(n):
    # 将数字转换为字符串以便获取每个位数
    str_n = str(n)
    # 获取数字的位数
    length = len(str_n)
    
    # 计算各个位数的立方和
    sum_of_cubes = sum(int(char)**length for char in str_n)
    
    # 如果和等于原数字，则它是水仙花数
    return n == sum_of_cubes

# 测试几个例子
numbers_to_check = [153, 370, 371, 407]
for num in numbers_to_check:
    print(f"{num} 是水仙花数吗? {is_narcissistic_number(num)}")