1. 程式人生 > 程式設計 >Python操作dict時避免出現KeyError的幾種解決方法

Python操作dict時避免出現KeyError的幾種解決方法

在讀取dict的key和value時,如果key不存在,就會觸發KeyError錯誤,如:

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

就會出現:

KeyError: 'd'

第一種解決方法

首先測試key是否存在,然後才進行下一步操作,如:

t = {
  'a': '1',}
if 'd' in t:
  print(t['d'])
else:
  print('not exist')

會出現:

not exist

第二種解決方法

利用dict內建的get(key[,default])方法,如果key存在,則返回其value,否則返回default;使用這個方法永遠不會觸發KeyError,如:

t = {
  'a': '1',}
print(t.get('d'))

會出現:

None

加上default引數:

t = {
  'a': '1',}
print(t.get('d','not exist'))
print(t)

會出現:

not exist
{'a': '1','b': '2'}

第三種解決方法

利用dict內建的setdefault(key[,default])方法,如果key存在,則返回其value;否則插入此key,其value為default,並返回default;使用這個方法也永遠不會觸發KeyError,如:

t = {
  'a': '1',}
print(t.setdefault('d'))
print(t)

會出現:

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

加上default引數:

t = {
  'a': '1',}
print(t.setdefault('d','not exist'))
print(t)

會出現:

not exist
{'c': '3','d': 'not exist','b': '2'}

第四種解決方法

向類dict增加__missing__()方法,當key不存在時,會轉向__missing__()方法處理,而不觸發KeyError,如:

t = {
  'a': '1',}

class Counter(dict):

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

會出現:

None

更改return值:

t = {
  'a': '1',key):
    return key
c = Counter(t)
print(c['d'])
print(c)

會出現:

d
{'c': '3','b': '2'}

第五種解決方法

利用collections.defaultdict([default_factory[,...]])物件,實際上這個是繼承自dict,而且實際也是用到的__missing__()方法,其default_factory引數就是向__missing__()方法傳遞的,不過使用起來更加順手:
如果default_factory為None,則與dict無區別,會觸發KeyError錯誤,如:

import collections
t = {
  'a': '1',}
t = collections.defaultdict(None,t)
print(t['d'])

會出現:

KeyError: 'd'

但如果真的想返回None也不是沒有辦法:

import collections
t = {
  'a': '1',}

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

會出現:

None

如果default_factory引數是某種資料型別,則會返回其預設值,如:

import collections
t = {
  'a': '1',}
t = collections.defaultdict(int,t)
print(t['d'])

會出現:

0

又如:

import collections
t = {
  'a': '1',}
t = collections.defaultdict(list,t)
print(t['d'])

會出現:

[]

注意:

如果dict內又含有dict,key巢狀獲取value時,如果中間某個key不存在,則上述方法均失效,一定會觸發KeyError:

import collections
t = {
  'a': '1',}
t = collections.defaultdict(dict,t)
print(t['d']['y'])

會出現:

KeyError: 'y'

到此這篇關於Python操作dict時避免出現KeyError的幾種解決方法的文章就介紹到這了,更多相關Python操作dict出現KeyError內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!