python 编写函数判断水仙花数

可以使用以下代码编写一个函数来判断一个数是否为水仙花数：

def is_armstrong_number(num):
    # 将数字转换为字符串
    num_str = str(num)
    # 初始化总和为0
    total = 0
    # 计算每个数字的n次方之和
    for digit in num_str:
        total += int(digit) ** len(num_str)
    # 判断是否为水仙花数
    if total == num:
        return True
    else:
        return False

函数的参数 `num` 为需要判断的数字，函数会将该数字转换为字符串，并逐个计算每个数字的 n 次方之和。如果计算出的总和等于原数，则说明该数字为水仙花数，函数会返回 `True`，否则返回 `False`。