在读取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'])
会出现:
- KeyError: 'y'