human
Program for Armstrong Numbers |"
How can the above be solved in Python?

=====
Give me a code to program for Armstrong Numbers and the other steps required?
assistant
#Here is code to program for Armstrong Numbers and the other steps required.
 def armstrong ( n ) :
	number = str ( n )
	n = len ( number )
	output = 0
	for i in number :
	output = output + int ( i ) ** n
	if output == int ( number ) :
		return ( True )
	else :
		return ( False )
print ( armstrong ( 153 ) )
print ( armstrong ( 120 ) )