Solución reto #15 Python

[grisales54]

numero = int(input("Digite un número: "))

def armstrong(numero):

lista = list(str(numero))

longi = len(lista)

num = 0

for j in range(len(lista)):

    num = num + (int(lista[j])) ** (longi)

if num == numero:
    return print("Número de armstrong")

else:
    return print("No es un número de armstrong")

armstrong(numero)