Írj egy függvényt, amely eldönti, hogy egy adott szám Armstrong-számnak számít-e!
153
def is_armstrong(szam):
    input_num = szam
    result = 0
    n = len(str(szam))
    while szam > 0:
        digit = szam % 10
        result += digit ** n
        szam //= 10
    return result == input_num