Python实现阿姆斯特朗数

如果一个n位正整数等于其各位数字的n次方之和,则称该数为阿姆斯特朗数。 例如1^3 + 5^3 + 3^3 = 153

def my_sum(num):
    """计算阿姆斯特朗数"""
    num_list = list(str(num)) # 转换为字符串列表
    n = len(num_list) # 计算长度
    my_list = [int(x)**n for x in num_list] # 计算
    s = sum(my_list)
    return s

li= []
# 计算10000内的阿姆斯朗数
for i in range(0, 10000):
    s1= my_sum(i)
    if i == s1:
        li.append(i)