Cartesian Product of any number of sets

Given N number of sets. The task is to write a program to perform cartesian product of all the sets in a given order.

Example:

Input:
1st set: 1 2
2nd set: A 
3rd set: x 
4th set: 5 6
Output:
[['1', 'A', 'x', '5'],
 ['1', 'A', 'x', '6'],
 ['2', 'A', 'x', '5'],
 ['2', 'A', 'x', '6']]

Input:
1st set: 1 2
2nd set: A 
3rd set: x y z 
Output:
[['1', 'A', 'x'],
 ['1', 'A', 'y'],
 ['1', 'A', 'z'],
 ['2', 'A', 'x'],
 ['2', 'A', 'y'],
 ['2', 'A', 'z']]

Approach: The approach is to compute the product of set-1 and set-2 at the beginning and then the resultant of set-1 and set-2 will have a product with set-3 and then the resultant of set-1, set-2, set-3 will have a Cartesian product with set-4 and so on till set-n.

Below is the implementation of the above approach.

filter_none

edit
close

play_arrow

link
brightness_4
code

# Python program for cartesian
# product of N-sets 
  
# function to find cartesian product of two sets 
def cartesianProduct(set_a, set_b):
    result =[]
    for i in range(0, len(set_a)):
        for j in range(0, len(set_b)):
  
            # for handling case having cartesian
            # prodct first time of two sets
            if type(set_a[i]) != list:         
                set_a[i] = [set_a[i]]
                  
            # coping all the members
            # of set_a to temp
            temp = [num for num in set_a[i]]
              
            # add member of set_b to 
            # temp to have cartesian product     
            temp.append(set_b[j])             
            result.append(temp)  
              
    return result
  
# Function to do a cartesian 
# product of N sets 
def Cartesian(list_a, n):
      
    # result of cartesian product
    # of all the sets taken two at a time
    temp = list_a[0]
      
    # do product of N sets 
    for i in range(1, n):
        temp = cartesianProduct(temp, list_a[i])
          
    print(temp)
  
# Driver Code
list_a = [[1, 2],          # set-1
          ['A'],          # set-2
          ['x', 'y', 'z']]   # set-3
            
# number of sets
n = len(list_a) 
  
# Function is called to perform 
# the cartesian product on list_a of size n 
Cartesian(list_a, n)

chevron_right


Output:

[[1, 'A', 'x'],
 [1, 'A', 'y'],
 [1, 'A', 'z'], 
 [2, 'A', 'x'], 
 [2, 'A', 'y'], 
 [2, 'A', 'z']]


My Personal Notes arrow_drop_up

Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.




Article Tags :

2


Please write to us at contribute@geeksforgeeks.org to report any issue with the above content.