# Python Armstrong number generator: Generate next Armstrong numbers

## Python: Generators Yield Exercise-14 with Solution

Write a Python program to implement a generator that generates the next Armstrong number after a given number.

When the sum of the cube of the individual digits of a number is equal to that number, the number is called Armstrong number. For Example 153 is an Armstrong number because 153 = 1^{3}+5^{3}+3^{3}.

**Sample Solution:**

**Python Code:**

```
def is_armstrong_number(n):
num_str = str(n)
num_len = len(num_str)
sum_of_cubes = sum(int(digit) ** num_len for digit in num_str)
return n == sum_of_cubes
def next_armstrong_number(start):
while True:
start += 1
if is_armstrong_number(start):
yield start
# Accept input from the user
n = int(input("Input a number: "))
# Create the generator object
armstrong_gen = next_armstrong_number(n)
# Generate and print the next Armstrong number
print("Next Armstrong number:")
for _ in range(1):
print(next(armstrong_gen))
```

Sample Output:

Input a number: 370 Next Armstrong number: 371

Enter a number: 153 Next Armstrong numbers: 370

Input a number: 407 Next Armstrong number: 1634

**Explanation:**

In the above exercise -

The is_armstrong_number() function first checks whether a given number is an Armstrong number. The function calculates the sum of cubes of digits and compares it with the original number to determine if it's an Armstrong number.

Next, the generator function next_armstrong_number() generates the next Armstrong numbers one by one.

**Flowchart:**

**Python Code Editor:**

** Contribute your code and comments through Disqus.**

**Previous:** Generate square, cube roots of numbers.

**Next:** Generate factors for a number.

**What is the difficulty level of this exercise?**

Test your Programming skills with w3resource's quiz.

**It will be nice if you may share this link in any developer community or anywhere else, from where other developers may find this content. Thanks.**

https://w3resource.com/python-exercises/generators-yield/python-generators-yield-exercise-14.php

**Weekly Trends and Language Statistics**- Weekly Trends and Language Statistics