快捷導(dǎo)航

Python操作dict時(shí)避免出現(xiàn)KeyError的幾種解決方法

更新時(shí)間：2020年09月20日 14:29:14 作者：風(fēng)景不在對(duì)岸wj

這篇文章主要介紹了Python操作dict時(shí)避免出現(xiàn)KeyError的幾種解決方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

在讀取dict的key和value時(shí)，如果key不存在，就會(huì)觸發(fā)KeyError錯(cuò)誤，如：

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

就會(huì)出現(xiàn)：

KeyError: 'd'

第一種解決方法

首先測(cè)試key是否存在，然后才進(jìn)行下一步操作，如：

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

會(huì)出現(xiàn)：

not exist

第二種解決方法

利用dict內(nèi)置的get(key[,default])方法，如果key存在，則返回其value,否則返回default;使用這個(gè)方法永遠(yuǎn)不會(huì)觸發(fā)KeyError，如：

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

會(huì)出現(xiàn)：

None

加上default參數(shù)：

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

會(huì)出現(xiàn)：

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

第三種解決方法

利用dict內(nèi)置的setdefault(key[,default])方法，如果key存在，則返回其value;否則插入此key，其value為default，并返回default;使用這個(gè)方法也永遠(yuǎn)不會(huì)觸發(fā)KeyError，如：

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

會(huì)出現(xiàn)：

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

加上default參數(shù)：

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

會(huì)出現(xiàn)：

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

第四種解決方法

向類dict增加__missing__()方法，當(dāng)key不存在時(shí)，會(huì)轉(zhuǎn)向__missing__()方法處理，而不觸發(fā)KeyError,如：

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

class Counter(dict):

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

會(huì)出現(xiàn)：

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)

會(huì)出現(xiàn)：

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

第五種解決方法

利用collections.defaultdict([default_factory[,...]])對(duì)象，實(shí)際上這個(gè)是繼承自dict，而且實(shí)際也是用到的__missing__()方法，其default_factory參數(shù)就是向__missing__()方法傳遞的，不過(guò)使用起來(lái)更加順手：
如果default_factory為None，則與dict無(wú)區(qū)別，會(huì)觸發(fā)KeyError錯(cuò)誤，如：

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

會(huì)出現(xiàn)：

KeyError: 'd'

但如果真的想返回None也不是沒(méi)有辦法：

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

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

會(huì)出現(xiàn)：

None

如果default_factory參數(shù)是某種數(shù)據(jù)類型，則會(huì)返回其默認(rèn)值，如：

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

會(huì)出現(xiàn)：

0

又如：

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

會(huì)出現(xiàn)：

[]

注意：

如果dict內(nèi)又含有dict，key嵌套獲取value時(shí)，如果中間某個(gè)key不存在，則上述方法均失效，一定會(huì)觸發(fā)KeyError：

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

會(huì)出現(xiàn)：

KeyError: 'y'

到此這篇關(guān)于Python操作dict時(shí)避免出現(xiàn)KeyError的幾種解決方法的文章就介紹到這了,更多相關(guān)Python操作dict出現(xiàn)KeyError內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: