Python : How to Remove multiple keys from Dictionary while Iterating ?
In this article we will discuss how to remove multiple key / value pairs from dictionary while iterating.
Suppose we have a dictionary of string and int i.e.
|
1 2 3 4 5 6 7 8 |
# Dictionary of strings wordFreqDic = { "Hello": 51, "at" : 23 , "test" : 33, "here" : 63, "start" : 13 } |
Now we want to remove all key/value pairs from dictionary whose value is divisible by 3. Let’s see how to do that,
Removing elements from Dictionary while Iterating
We can not change the size of dictionary while iterating over it. If we try to remove any element from dictionary while iterating over it then it will throw error.
For example,
|
1 2 3 |
for (key, value) in wordFreqDic.items() : if value % 3 == 0: del wordFreqDic[key] |
Will through following Error,
|
1 |
RuntimeError: dictionary changed size during iteration |
So, let’s see how to delete multiple elements from dictionary.
Removing keys from dictionary while iterating by creating a list of keys
As we cannot iterate and remove elements from dictionary at same time. Therefore will, first iterate over the dictionary and create a list of keys that need to be deleted from dictionary. Now, we will iterate over this list of keys and delete their entries from dictionary i.e.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
''' Removing multiple keys from dictionary by creating a list of keys to be deleted and then deleting them one by one ''' # List of keys to be deleted from dictionary selectedKeys = list() # Iterate over the dict and put to be deleted keys in the list for (key, value) in wordFreqDic.items() : if value % 3 == 0: selectedKeys.append(key) # Iterate over the list and delete corresponding key from dictionary for key in selectedKeys: if key in wordFreqDic : del wordFreqDic[key] |
Now dictionary contents will be,
|
1 |
Modified Dictionary {'start': 13, 'at': 23} |
Removing keys from dictionary while iterating by creating a copy of dictionary
Create a copy of dictionary and iterate over this temporary dictionary, check element to be deleted and delete it from original dictionary i.e.
|
1 2 3 4 5 6 7 8 9 10 11 12 |
''' Removing multiple keys from dictionary by creating a copy of dictionary and iterating over it ''' # Create a temporary copy of dictionary copyOfDict = dict(wordFreqDic) # Iterate over the temporary dictionary and delete corresponding key from original dictionary for (key, value) in copyOfDict.items() : if value % 3 == 0: del wordFreqDic[key] |
Now dictionary contents will be,
|
1 |
Modified Dictionary {'start': 13, 'at': 23} |
Complete example is as follows,
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
def main(): # Dictionary of strings wordFreqDic = { "Hello": 51, "at" : 23 , "test" : 33, "here" : 63, "start" : 13 } print("Original Dictionary ", wordFreqDic) ''' Removing multiple keys from dictionary by creating a list of keys to be deleted and then deleting them one by one ''' # List of keys to be deleted from dictionary selectedKeys = list() # Iterate over the dict and put to be deleted keys in the list for (key, value) in wordFreqDic.items() : if value % 3 == 0: selectedKeys.append(key) # Iterate over the list and delete corresponding key from dictionary for key in selectedKeys: if key in wordFreqDic : del wordFreqDic[key] print("Modified Dictionary " , wordFreqDic) # Dictionary of strings wordFreqDic = { "Hello": 51, "at" : 23 , "test" : 33, "here" : 63, "start" : 13 } print("Original Dictionary ", wordFreqDic) ''' Removing multiple keys from dictionary by creating a copy of dictionary and iterating over it ''' # Create a temporary copy of dictionary copyOfDict = dict(wordFreqDic) # Iterate over the temporary dictionary and delete corresponding key from original dictionary for (key, value) in copyOfDict.items() : if value % 3 == 0: del wordFreqDic[key] print("Modified Dictionary ", wordFreqDic) if __name__ == '__main__': main() |
Output:
|
1 2 3 4 |
Original Dictionary {'Hello': 51, 'test': 33, 'here': 63, 'start': 13, 'at': 23} Modified Dictionary {'start': 13, 'at': 23} Original Dictionary {'Hello': 51, 'test': 33, 'here': 63, 'start': 13, 'at': 23} Modified Dictionary {'start': 13, 'at': 23} |
Follow @thisPtr
Leave a Reply