The simple and elegant way to express multiple for loop in python?

Question

I have below code, would like to know what is the best simple and elegant way to express multiple for loop?

for x in range (10):
    for y in range (10):
        for z in range(10):
            if x+y+z=10:
                print (x,y,z)

Thanks in advance!

There Are solutions.But This code is better readable.

Amin Golmahalleh
– Amin Golmahalleh

2019-12-07 14:15:18 +00:00
Commented Dec 7, 2019 at 14:15 — Amin Golmahalleh
– Amin Golmahalleh, Commented Dec 7, 2019 at 14:15

RafalS · Accepted Answer · 2019-12-07 14:14:29Z

5

from itertools import product

for x, y, z in product(range(10), range(10), range(10)):
    if x + y + z == 10:
        print(x, y, z)

To remove range(10) duplication use this:

for x, y, z in product(range(10), repeat=3):

EDIT: as Tomerikoo pointed out - in this specific case the code will be more flexible if you don't unpack the tuple:

for numbers in product(range(10), repeat=3):
    if sum(numbers) == 10:
        print(*numbers)

edited Dec 7, 2019 at 14:14

answered Dec 7, 2019 at 13:49

RafalS

6,5441 gold badge23 silver badges26 bronze badges

Sign up to request clarification or add additional context in comments.

6 Comments

chepner Over a year ago

product(range(10), times=3). The whole point of the times argument is to avoid repeating the range you want to use multiple times.

Tomerikoo Over a year ago

to make it more generic regardless of number of loops can be: for prod in product(range(10), times=n): if sum(prod) == 10: print. where n is the number of loops

chepner Over a year ago

@Tomerikoo No; again, product has a times argument to avoid that kind of convoluted argument construction.

Tomerikoo Over a year ago

I get: TypeError: 'times' is an invalid keyword argument for product() maybe you meant repeat

Tomerikoo Over a year ago

@chepner what if now you want to check it for 2 or 4 numbers? you need to go and add/remove variables and change the if expression. My suggestion simply makes it without the use of variables to better handle future changes

|

Collectives™ on Stack Overflow

The simple and elegant way to express multiple for loop in python?

1 Answer 1

6 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

6 Comments

Your Answer

Sign up or log in

Post as a guest

Related