Python:操作dict时避免出现KeyError的几种方法

在读取dict的key和value时，如果key不存在，就会触发KeyError错误，如：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
print(t["d"])

就会出现：

KeyError: "d"

首先测试key是否存在，然后才进行下一步操作，如：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
if "d" in t:
    print(t["d"])
else:
    print("not exist")

会出现：

not exist

利用dict内置的get(key[,default])方法，如果key存在，则返回其value,否则返回default;使用这个方法永远不会触发KeyError，如：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
print(t.get("d"))

会出现：

None

加上default参数：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
print(t.get("d", "not exist"))
print(t)

会出现：

not exist
{"a": "1", "c": "3", "b": "2"}

利用dict内置的setdefault(key[,default])方法，如果key存在，则返回其value;否则插入此key，其value为default，并返回default;使用这个方法也永远不会触发KeyError，如：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
print(t.setdefault("d"))
print(t)

会出现：

None
{"b": "2", "d": None, "a": "1", "c": "3"}

加上default参数：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
print(t.setdefault("d", "not exist"))
print(t)

会出现：

not exist
{"c": "3", "d": "not exist", "a": "1", "b": "2"}

向类dict增加__missing__()方法，当key不存在时，会转向__missing__()方法处理，而不触发KeyError,如：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}


class Counter(dict):

    def __missing__(self, key):
        return None
c = Counter(t)
print(c["d"])

会出现：

None

更改return值：

Python

t = {
    "a": "1",
    "b": "2",
    "c": "3",
}


class Counter(dict):

    def __missing__(self, key):
        return key
c = Counter(t)
print(c["d"])
print(c)

会出现：

d
{"c": "3", "a": "1", "b": "2"}

利用collections.defaultdict([default_factory[,...]])对象，实际上这个是继承自dict，而且实际也是用到的__missing__()方法，其default_factory参数就是向__missing__()方法传递的，不过使用起来更加顺手：
如果default_factory为None，则与dict无区别，会触发KeyError错误，如：

Python

import collections
t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
t = collections.defaultdict(None, t)
print(t["d"])

会出现：

KeyError: "d"

但如果真的想返回None也不是没有办法：

Python

import collections
t = {
    "a": "1",
    "b": "2",
    "c": "3",
}

def handle():
    return None
t = collections.defaultdict(handle, t)
print(t["d"])

会出现：

None

如果default_factory参数是某种数据类型，则会返回其默认值，如：

Python

import collections
t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
t = collections.defaultdict(int, t)
print(t["d"])

会出现：

0

又如：

Python

import collections
t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
t = collections.defaultdict(list, t)
print(t["d"])

会出现：

[]

---

注意：
如果dict内又含有dict，key嵌套获取value时，如果中间某个key不存在，则上述方法均失效，一定会触发KeyError：

Python

import collections
t = {
    "a": "1",
    "b": "2",
    "c": "3",
}
t = collections.defaultdict(dict, t)
print(t["d"]["y"])

会出现：

KeyError: "y"