CODEWITHSUNDEEP
python
OneMoreError
OneMoreError

Reputation: 7728

Removing entries from a dictionary based on values

I have a dictionary with character-integer key-value pair. I want to remove all those key value pairs where the value is 0.

For example:

>>> hand
{'a': 0, 'i': 0, 'm': 1, 'l': 1, 'q': 0, 'u': 0}

I want to reduce the same dictionary to this:

>>> hand
{'m': 1, 'l': 1}

Is there an easy way to do that?

Upvotes: 47

Views: 41700

Answers (4)

Fabian
Fabian

Reputation: 4348

If you don't want to create a new dictionary, you can use this:

>>> hand = {'a': 0, 'i': 0, 'm': 1, 'l': 1, 'q': 0, 'u': 0}
>>> for key in list(hand.keys()):  ## creates a list of all keys
...     if hand[key] == 0:
...             del hand[key]
... 
>>> hand
{'m': 1, 'l': 1}
>>>

Upvotes: 13

Niklas B.
Niklas B.

Reputation: 95298

You can use a dict comprehension:

>>> { k:v for k, v in hand.items() if v }
{'m': 1, 'l': 1}

Or, in pre-2.7 Python, the dict constructor in combination with a generator expression:

>>> dict((k, v) for k, v in hand.iteritems() if v)
{'m': 1, 'l': 1}

Upvotes: 39

Volatility
Volatility

Reputation: 32300

hand = {k: v for k, v in hand.iteritems() if v != 0}

For Pre-Python 2.7:

hand = dict((k, v) for k, v in hand.iteritems() if v != 0)

In both cases you're filtering out the keys whose values are 0, and assigning hand to the new dictionary.

Upvotes: 11

Pavel Anossov
Pavel Anossov

Reputation: 62898

A dict comprehension?

{k: v for k, v in hand.items() if v != 0}

In python 2.6 and earlier:

dict((k, v) for k, v in hand.items() if v != 0)

Upvotes: 4

Related Questions