Python操作dict时避免出现KeyError的几种解决方法_Python

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

2020-09-21 00:03风景不在对岸wj Python

这篇文章主要介绍了Python操作dict时避免出现KeyError的几种解决方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

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

				?

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									print(t['d'])

就会出现：

KeyError: 'd'

第一种解决方法

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

				?

									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，如：

				?

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									print(t.get('d'))

会出现：

None

加上default参数：

				?

									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，如：

				?

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									print(t.setdefault('d'))

									print(t)

会出现：

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

加上default参数：

				?

									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,如：

				?

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									class Counter(dict):

									  def __missing__(self, key):

									    return None

									c = Counter(t)

									print(c['d'])

会出现：

None

更改return值：

				?

									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错误，如：

				?

									import collections

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									t = collections.defaultdict(None, t)

									print(t['d'])

会出现：

KeyError: 'd'

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

				?

									import collections

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									def handle():

									  return None

									t = collections.defaultdict(handle, t)

									print(t['d'])

会出现：

None

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

				?

									import collections

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									t = collections.defaultdict(int, t)

									print(t['d'])

会出现：

0

又如：

				?

									import collections

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									t = collections.defaultdict(list, t)

									print(t['d'])

会出现：

[]

注意：

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

				?

									import collections

									t = {

									  'a': '1',

									  'b': '2',

									  'c': '3',

									}

									t = collections.defaultdict(dict, t)

									print(t['d']['y'])