Írj egy Python függvényt, amely meghatározza, hogy egy adott szám Armstrong-szám-e (amikor az eredmény az adott számjegyek köbének összege)!
371
def is_armstrong(number):
    sum = 0
    temp = number
    while temp > 0:
        digit = temp % 10
        sum += digit ** 3
        temp //= 10
    return number == sum

print(is_armstrong(371))