Question

Suppose I have the following code to collect all the possible combinations.

abArray = []
for a in range(minA, maxA, aStep):
  for b in range(minB, maxB, bStep):
    if a < b: continue
    abArray.append((a,b))

Is there a more efficient way to store all possible combinations with a condition rather than using nested for loops?

Answer 1

I think you’re looking for itertools.product:

from itertools import product
abGenerator = product(range(minA, maxA, stepA), range(minB, maxB, stepB))
abArray = (i for i in abGenerator if i[0] < i[1])