Python program to apply itertools.product to elements of a list of lists

Itertools is a module that consists of the methods to apply various iteration based operations including combinations, permutations, etc., on the iterable components in Python. It has a set lightweight, memory-efficient and fast tools for performing iterator algebra.

Note: For more information, refer to Python Itertools

itertools.product()

It is used to perform cartesian product within a list or among lists. The nested loops cycle in a way that the rightmost element advancing on every iteration. This pattern creates a lexicographic ordering and thus if the input’s iterables are sorted, the product tuples are also in sorted order.

It takes iterables as the parameter. The below example shows a very simple representation of itertools.product() method. Here it is used as a creation of a cartesian product.

Example:

Output:

[(‘G’, ‘G’), (‘G’, ‘F’), (‘G’, ‘G’), (‘f’, ‘G’), (‘f’, ‘F’), (‘f’, ‘G’), (‘G’, ‘G’), (‘G’, ‘F’), (‘G’, ‘G’)]

Operating on list of lists

To use itertools.product() method on list of lists, perform unpacking operation first. It can be done using two ways:

• By unpacking the list inside function

  The example below shows that how can unpacking be performed by simple operation within the method.

  
  

  
  
  

  import itertools       def product(list_of_str):            str1 = list_of_str[0]     str2 = list_of_str[1]            # returning the list      # containing cartesian product     return [x for x in itertools.product(list(str1),                                          list(str2))]    print(product(["GfG", "GFG"]))

  Output

  [(‘G’, ‘G’), (‘G’, ‘F’), (‘G’, ‘G’), (‘f’, ‘G’), (‘f’, ‘F’), (‘f’, ‘G’), (‘G’, ‘G’), (‘G’, ‘F’), (‘G’, ‘G’)]

  The disadvantage of this way is that, it requires additional information to be known i.e the length of the list inside the lists.

• Using ‘*’ operator

  To overcome the above mentioned disadvantage ‘*’ is used to unpack the lists within the list. So the above code can be optimized as follows:

  
  

  
  
  

  import itertools       def product(lst):            # Unpack operation performed     # by '*' operator and returning     # the list containing cartesian     # product     return [x for x in itertools.product(*lst)]       # list of lists being passed in the method print(product(["GfG", "GFG"]))

  Output

  [(‘G’, ‘G’), (‘G’, ‘F’), (‘G’, ‘G’), (‘f’, ‘G’), (‘f’, ‘F’), (‘f’, ‘G’), (‘G’, ‘G’), (‘G’, ‘F’), (‘G’, ‘G’)]