Create P81_armstrong number.py

Author: pmishra98

Programs/P81_armstrong number.py

@@ -0,0 +1,31 @@
+# Python program to check if the number is an Armstrong number or not
+
+# take input from the user
+
+num = int(input("Enter a number: "))
+
+# initialize sum
+
+sum = 0
+
+# find the sum of the cube of each digit
+
+temp = num
+
+while temp > 0:
+
+   digit = temp % 10
+
+   sum += digit ** 3
+
+   temp //= 10
+
+# display the result
+
+if num == sum:
+
+   print(num,"is an Armstrong number")
+
+else:
+
+   print(num,"is not an Armstrong number")