我需要一种方法来获得字典值,如果它的键存在,或简单地返回None,如果它不存在。

然而,如果你搜索一个不存在的键,Python会引发一个KeyError异常。我知道我可以检查钥匙,但我在寻找更明确的东西。如果键不存在,是否有一种方法只返回None ?


当前回答

您可以使用dict对象的get()方法,正如其他人已经建议的那样。或者,根据你正在做的事情,你可以使用try/except套件,像这样:

try:
   <to do something with d[key]>
except KeyError:
   <deal with it not being there>

这被认为是处理这种情况的一种非常“python”的方法。

其他回答

给那些使用字典的人。获取嵌套字典的技术,而不是显式地检查字典的每一层,或扩展dict类,您可以将默认返回值设置为除最外层之外的空字典。这里有一个例子:

my_dict = {'level_1': {
             'level_2': {
                  'level_3': 'more_data'
                  }
              }
           }
result = my_dict.get('level_1', {}).get('level_2', {}).get('level_3')
# result -> 'more_data'
none_result = my_dict.get('level_1', {}).get('what_level', {}).get('level_3')
# none_result -> None

警告:请注意,此技术仅在预期键值为字典时有效。如果键what_level确实存在于字典中,但它的值是字符串或整数等,那么它将引发AttributeError。

如果你有一个更复杂的需求,等价于缓存,这个类可能会派上用场:

class Cache(dict):
    """ Provide a dictionary based cache

        Pass a function to the constructor that accepts a key and returns
        a value.  This function will be called exactly once for any key
        required of the cache.
    """

    def __init__(self, fn):
        super()
        self._fn = fn

    def __getitem__(self, key):
        try:
            return super().__getitem__(key)
        except KeyError:
            value = self[key] = self._fn(key)
            return value

构造函数接受一个用键调用的函数,该函数应返回字典的值。然后存储这个值,并在下次从字典中检索。像这样使用它……

def get_from_database(name):
    # Do expensive thing to retrieve the value from somewhere
    return value

answer = Cache(get_from_database)
x = answer(42)   # Gets the value from the database
x = answer(42)   # Gets the value directly from the dictionary

我通常在这种情况下使用defaultdict。您提供一个不接受参数的工厂方法,并在看到新键时创建一个值。当您希望在新键上返回一个空列表之类的内容时,它会更有用(请参阅示例)。

from collections import defaultdict
d = defaultdict(lambda: None)
print d['new_key']  # prints 'None'

你可以使用dict.get()

value = d.get(key)

如果key不在d中,返回None。你也可以提供一个不同的默认值来代替None:

value = d.get(key, "empty")

如果你想要一个更透明的解决方案,你可以子类dict来获得这个行为:

class NoneDict(dict):
    def __getitem__(self, key):
        return dict.get(self, key)

>>> foo = NoneDict([(1,"asdf"), (2,"qwerty")])
>>> foo[1]
'asdf'
>>> foo[2]
'qwerty'
>>> foo[3] is None
True