how to avoid nested loop in python and check condition on all combination of elements in multi list

Question

I do have the following n=3 list (nested loops) I need it to be general and works for any n lists

object1 = [4, 5, 3]
object2 = [1, 2, 3, 4]
object3 = [1, 0]

for c1 in object1:
    for c2 in object2:
        for c3 in object3:
            if c1+c2+c3<9:
                print(c1,c2,c3)

How can I do it in an efficient way in Python?

score 3 · Accepted Answer · answered Jun 29 '22 at 08:01

You need itertools.product:

import itertools
for cs in itertools.product(*[object1, object2, object3]):
    if sum(cs) < 9:
        print(cs[0], cs[1], cs[2])

# OR
# for c1, c2, c3 in itertools.product(*[object1, object2, object3]):
#     if (c1+c2+c3) < 9:
#         print(c1,c2,c3)

Output:

how to avoid nested loop in python and check condition on all combination of elements in multi list

1 Answers1