Python | Intersection of two nested list
Last Updated :
15 May, 2023
This particular article aims at achieving the task of intersecting two list, in which each element is in itself a list. This is also a useful utility as this kind of task can come in life of programmer if he is in the world of development. Lets discuss some ways to achieve this task.
Method 1: Naive Method This is the simplest method to achieve this task and uses the brute force approach of executing a loop and to check if one list contains similar list as of the other list.
Python3
test_list1 = [ [1, 2], [3, 4], [5, 6] ]
test_list2 = [ [3, 4], [5, 7], [1, 2] ]
print ("The original list 1 : " + str(test_list1))
print ("The original list 2 : " + str(test_list2))
res_list = []
for i in test_list1:
if i in test_list2:
res_list.append(i)
print ("The intersection of two lists is : " + str(res_list))
|
Output :
The original list 1 : [[1, 2], [3, 4], [5, 6]]
The original list 2 : [[3, 4], [5, 7], [1, 2]]
The intersection of two lists is : [[1, 2], [3, 4]]
Time Complexity: O(n2)
Auxiliary Space: O(n)
Method 2: Using list comprehension We can perform the similar task as discussed above in comparatively less no. of lines. But the method is similar as above just uses list comprehension approach to perform the task.
Python3
test_list1 = [ [1, 2], [3, 4], [5, 6] ]
test_list2 = [ [3, 4], [5, 7], [1, 2] ]
print ("The original list 1 : " + str(test_list1))
print ("The original list 2 : " + str(test_list2))
res_list = [i for i in test_list1 if i in test_list2]
print ("The intersection of two lists is : " + str(res_list))
|
Output :
The original list 1 : [[1, 2], [3, 4], [5, 6]]
The original list 2 : [[3, 4], [5, 7], [1, 2]]
The intersection of two lists is : [[1, 2], [3, 4]]
Time Complexity: O(n2)
Auxiliary Space: O(n)
Method 3: Using set() + map() and & The most efficient and recommended method to perform this task is using the combination of set() and map() to achieve it. Firstly converting inner lists to tuples using map, and outer lists to set, use of & operator can perform the set intersection and hence perform this task. Further if it is required to get in lists of list fashion, we can convert outer and inner containers back to list using map().
Python3
test_list1 = [ [1, 2], [3, 4], [5, 6] ]
test_list2 = [ [3, 4], [5, 7], [1, 2] ]
print ("The original list 1 : " + str(test_list1))
print ("The original list 2 : " + str(test_list2))
res_set = set(map(tuple, test_list1)) & set(map(tuple, test_list2))
res_list = list(map(list, res_set))
print ("The intersection of two lists is : " + str(res_list))
|
Output :
The original list 1 : [[1, 2], [3, 4], [5, 6]]
The original list 2 : [[3, 4], [5, 7], [1, 2]]
The intersection of two lists is : [[1, 2], [3, 4]]
Time Complexity: O(n)
Auxiliary Space: O(n)
Method 4: Using filter() and lambda function
- Convert both nested lists into sets using map() and set().
- Create a lambda function that takes a nested list and returns True if it’s a subset of the second set.
- Use filter() with the lambda function to get a filter object of the subsets.
- Convert the filter object to a list using list().
- Print the resulting list.
Python3
test_list1 = [ [1, 2], [3, 4], [5, 6] ]
test_list2 = [ [3, 4], [5, 7], [1, 2] ]
print ("The original list 1 : " + str(test_list1))
print ("The original list 2 : " + str(test_list2))
set1 = set(map(tuple, test_list1))
set2 = set(map(tuple, test_list2))
intersection_list = list(filter(lambda x: tuple(x) in set2, test_list1))
print ("The intersection of two lists is : " + str(intersection_list))
|
Output
The original list 1 : [[1, 2], [3, 4], [5, 6]]
The original list 2 : [[3, 4], [5, 7], [1, 2]]
The intersection of two lists is : [[1, 2], [3, 4]]
Time Complexity: O(n^2), where n is the length of the nested lists.
Auxiliary Space: O(k), where k is the size of the intersection list.
Please Login to comment...