CODEWITHSUNDEEP
pythondictionary
Anne Aunyme
Anne Aunyme

Reputation: 506

Bast way to update a value in a dict

I have a dict that contains parameters. I want to consider that any unspecified parameter should be read as a zero. I see several ways of doing it and wonder which one is recommended:

parameters = {
    "apples": 2
}

def gain_fruits0 (quantity, fruit):
    if not fruit in parameters :
        parameters[fruit] = 0
    parameters[fruit] += quantity

def gain_fruits1 (quantity, fruits):
    parameters[fruit] = quantity + parameters.get(fruit,0)

parameters is actually way bigger than that, if that is important to know for optimization purposes.

So, what would be the best way? gain_fruits0, gain_fruits1, or something else?

Upvotes: 0

Views: 41

Answers (1)

Seb
Seb

Reputation: 4576

This is a typical use of defaultdict, which works exactly like a regular dictionary except that it has the functionality you're after built in:

>>> from collections import defaultdict
>>> d = defaultdict(int)  # specify default value `int()`, which is 0
>>> d['apples'] += 1
>>> d
defaultdict(int, {'apples': 1})
>>> d['apples']  # can index normally
1
>>> d['oranges']  # missing keys map to the default value
0
>>> dict(d)  # can also cast to regular dict
{'apples': 1, 'oranges': 0}

Upvotes: 3

Related Questions