快捷導(dǎo)航

解決Python字典查找報(bào)Keyerror的問(wèn)題

更新時(shí)間：2021年05月26日 15:43:58 作者：幸運(yùn)的Alina

這篇文章主要介紹了解決Python字典查找報(bào)Keyerror的問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

Python的字典一般都直接查找key ，比如

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

但是如果在查找的key不存在的時(shí)候就會(huì)報(bào)：KeyError:

比如你要查看print(dict['d'])

由于這個(gè)時(shí)候dict里面并沒(méi)有這個(gè)key ,所以就會(huì)直接報(bào)錯(cuò)，那么這個(gè)時(shí)候其實(shí)python給我們提供了一種很棒的解決方法，那就是用

setdefault,用法如下： dict.setdefault(key,[這里設(shè)置如果不存在想將值設(shè)置為啥，默認(rèn)為None])

那么這里我們可以用此方法解決：

print(dict.setdefault('d',0))

然后就沒(méi)有問(wèn)題啦,注意下就是setdefault是如果要想往dict中添加新值的時(shí)候就使用此函數(shù)，如果只是單純想要做查找，碰到key不存在或者希望在通過(guò)這個(gè)鍵讀取值的時(shí)候能得到一個(gè)默認(rèn)值，那么建議用defaultdict

首先先介紹下這個(gè)所謂的defaultdict,來(lái)自于collections模塊，collections是個(gè)集合模塊，defaultdict(function_factory)構(gòu)建的是一個(gè)類似dictionary的對(duì)象，其中key的值，自行確定賦值，但是value的類型是function_factory的類實(shí)例，而且具有默認(rèn)值.這里還引入了一個(gè)概念就是工廠函數(shù)，python的工廠函數(shù)就是指那些內(nèi)建函數(shù)都是類對(duì)象，當(dāng)你調(diào)用他們時(shí)，實(shí)際上是創(chuàng)建了一個(gè)類實(shí)例。

比如int(),str(),set()等，這里我們看下例子：

import collections
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
d = collections.defaultdict(list)
for k, v in s:
    d[k].append(v)
print(d['yellow'])
print(d['white'])
print(list(d.items()))

我們最后得到的輸出結(jié)果如下：

我們可以看到，當(dāng)d中沒(méi)有對(duì)應(yīng)的key時(shí)最后返回的是個(gè)空列表，那是因?yàn)槲覀冊(cè)谠O(shè)置defaultdict的時(shí)候用的工廠函數(shù)是list,而list的默認(rèn)值是空列表，下面我們?cè)诳聪氯绻S函數(shù)是set()會(huì)是什么樣子

import collections
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
d = collections.defaultdict(set)
for k, v in s:
    d[k].add(v)
print(d['yellow'])
print(d['white'])
print(list(d.items()))

結(jié)果輸出如下：