python - 为 dict 和 defaultdict 编写代码

Question

我有以下问题:

from collections import defaultdict

def give_it_to_me(d):
    # This will crash if key 'it' does not exist
    return d['it']

def give_it2_to_me(d):
    # This will never crash, but does not trigger the default value in defaultdict
    return d.get('it2')

d1 = defaultdict(int)
d2 = { 'it' : 55 }

print give_it_to_me(d1)
print give_it_to_me(d2)

print give_it2_to_me(d1)
print give_it2_to_me(d2)

正如您在评论中看到的那样，编写一个版本的 give_it_to_me 似乎是不可能的:

1. 永远不会崩溃
2. 触发defaultdict的默认值

还是我错了？

Answer 1

您可能需要在 give_it_to_me 函数中添加更多代码。

使用 try except 语句检查现有 key 。

例如:

def give_it_to_me(d):
    # This won't crash if key 'it' does not exist in a normal dict.
    # It returns just None in that case.
    # It returns the default value of an defaultdict if the key is not found.
    try:
        return d['it']
    except KeyError:
        return None