Python | Filter dictionary of tuples by condition
Last Updated :
05 May, 2023
Sometimes, we can have a very specific problem in which we are given a tuple pair as values in dictionary and we need to filter dictionary items according to those pairs. This particular problem as use case in Many geometry algorithms in competitive programming. Let’s discuss certain ways in which this task can be performed.
Method #1 : Using items() + dictionary comprehension These functionalities together can achieve this task. We can access all values using items() and condition can be imposed by dictionary comprehension.
Python3
test_dict = { 'a' : ( 6 , 3 ), 'b' : ( 4 , 8 ), 'c' : ( 8 , 4 )}
print ("The original dictionary is : " + str (test_dict))
res = {key: val for key, val in test_dict.items() if val[ 0 ] > = 6 and val[ 1 ] < = 4 }
print ("The filtered dictionary is : " + str (res))
|
Time Complexity: O(n)
Space Complexity: O(n)
Output :
The original dictionary is : {'b': (4, 8), 'a': (6, 3), 'c': (8, 4)}
The filtered dictionary is : {'a': (6, 3), 'c': (8, 4)}
Method #2 : Using lambda + filter() This method works in similar way as above method with the exception of using the filter function instead of dictionary comprehension to compact logic. Works only with Python2.
Python
test_dict = { 'a' : ( 6 , 3 ), 'b' : ( 4 , 8 ), 'c' : ( 8 , 4 )}
print ("The original dictionary is : " + str (test_dict))
res = dict ( filter ( lambda (x, (y, z)): y > = 6 and z < = 4 , test_dict.items()))
print ("The filtered dictionary is : " + str (res))
|
Time Complexity: O(n)
Space Complexity: O(n)
Output :
The original dictionary is : {'b': (4, 8), 'a': (6, 3), 'c': (8, 4)}
The filtered dictionary is : {'a': (6, 3), 'c': (8, 4)}
Method 3 : Using a for loop
step by step approach :
Create a dictionary called test_dict with 3 key-value pairs, where each value is a tuple containing 2 integers.
Print the original dictionary using the print() function and string concatenation.
Create an empty dictionary called new_dict.
Use a for loop to iterate through the items in test_dict.
For each item, check if the first value in the tuple is greater than or equal to 6 and the second value in the tuple is less than or equal to 4.
If the condition is true, add the item to new_dict.
Print the filtered dictionary using the print() function and string concatenation.
I hope that helps!
Python3
test_dict = { 'a' : ( 6 , 3 ), 'b' : ( 4 , 8 ), 'c' : ( 8 , 4 )}
print ( "The original dictionary is: " + str (test_dict))
new_dict = {}
for key, value in test_dict.items():
if value[ 0 ] > = 6 and value[ 1 ] < = 4 :
new_dict[key] = value
print ( "The filtered dictionary is: " + str (new_dict))
|
Output
The original dictionary is: {'a': (6, 3), 'b': (4, 8), 'c': (8, 4)}
The filtered dictionary is: {'a': (6, 3), 'c': (8, 4)}
The time complexity of this method is O(n), where n is the number of items in the dictionary.
The auxiliary space is also O(n), as we create a new dictionary to store the filtered items.
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...