阿姆斯特朗数python

python实现阿姆斯特朗数（又称水仙花数）

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

def amstl(n):
    sum = 0
    num = len(str(n))
    temp = n
    while temp > 0:
        ams = n % 10
        sum += ams ** num
        temp //= 10


    if sum == n:
        return True
    else:
        return False