How to get the cartesian product of a series of lists with Python?

Sometimes, we want to get the cartesian product of a series of lists with Python.

In this article, we’ll look at how to get the cartesian product of a series of lists with Python.

How to get the cartesian product of a series of lists with Python?

To get the cartesian product of a series of lists with Python, we can use the itertools.product method.

For instance, we write:

import itertools

some_lists = [
   [1, 2, 3],
   ['a', 'b'],
   [4, 5]
]
for element in itertools.product(*some_lists):
    print(element)

We call itertools.product with the lists in some_lists spread into it as arguments.

Then we print all the tuples that are in the cartesian products list.

Therefore, we get:

(1, 'a', 4)
(1, 'a', 5)
(1, 'b', 4)
(1, 'b', 5)
(2, 'a', 4)
(2, 'a', 5)
(2, 'b', 4)
(2, 'b', 5)
(3, 'a', 4)
(3, 'a', 5)
(3, 'b', 4)
(3, 'b', 5)

printed.

Conclusion

To get the cartesian product of a series of lists with Python, we can use the itertools.product method.