Python嵌套字典查找使用默认值

时间:2010-08-09 06:14:58

标签: python dictionary default-value lookup

>>> d2
{'egg': 3, 'ham': {'grill': 4, 'fry': 6, 'bake': 5}, 'spam': 2}
>>> d2.get('spamx',99)
99
>>> d2.get('ham')['fry']
6

我想获得火腿里面的鱼苗值,如果没有,得到值,99或88作为第二个例子。但是如何?

4 个答案:

答案 0 :(得分:13)

d2.get('ham', {}).get('fry', 88)

我可能会将其分解为现实生活中的几个陈述。

ham = d2.get('ham', {})
fry = ham.get('fry', 88)

答案 1 :(得分:3)

对于get的正常工作的默认值,第一个默认值需要是字典,这样如果第一次失败,你可以正确地链接.get调用。

d.get('ham',{}).get('fry',88)

除了块

之外,您还可以使用try
def get_ham_fry()
  try:
    return d['ham']['fry']
  except AttributeError,e:
    return 88

答案 2 :(得分:3)

如果你需要做很多事情,你可以写一个辅助函数

def get_nested(d, list_of_keys, default):
    for k in list_of_keys:
        if k not in d: 
            return default
        d=d[k]
    return d

print get_nested(d2,['ham','spam'],99)
print get_nested(d2,['ham','grill'],99)

答案 3 :(得分:1)

这是处理嵌套字典的解决方案:

def get(root, *keys):
    """
    Returns root[k_1][k_2]...[k_n] if all k_1, ..., k_n are valid keys/indices. 
    Returns None otherwise
    """
    if not keys:
        return root
    if keys[0] not in root:
        return None
    if keys[0] in root:
        return get(root[keys[0]], *keys[1:])

用法:

>>> d = {'a': 1, 'b': {'c': 3}}
>>> get(d, 'b', 'c')
3
>>> get(d. 'key that's not in d')
None
>>> get(d)
{'a': 1, 'b': {'c': 3}}