how to find armstrong number in python