字典(dict)對(duì)象是Python最常用的數(shù)據(jù)結(jié)構(gòu),社區(qū)曾有人開玩笑地說:”Python企圖用字典裝載整個(gè)世界”,字典在Python中的重要性不言而喻,這里整理了幾個(gè)關(guān)于高效使用字典的清單,助你代碼更加Pythonic。
用in關(guān)鍵字檢查key是否存在
Python之禪中有一條開發(fā)哲學(xué)是:
Thereshouldbeone—andpreferablyonlyone—obviouswaytodoit.
盡量找一種,最好是唯一種顯而易見的解決方案。Python2中判斷某個(gè)key是否存在字典中可使用has_key方法,另外一種方式是使用in關(guān)鍵字。但是強(qiáng)烈推薦使用后者,因?yàn)閕n的處理速度更快,另外一個(gè)原因是has_key這個(gè)方法在Python3被移除了,要想同時(shí)兼容py2和py3兩個(gè)版本的代碼,用in是最好的選擇。
bad
d={'name':'python'}
ifd.has_key('name'):
pass
good
if'name'ind:
pass
用get獲取字典中的值
關(guān)于獲取字典中的值,一種簡(jiǎn)單的方式就是用d.[x]訪問該元素,但是這種情況在key不存在的情況下會(huì)報(bào)KeyError錯(cuò)誤,當(dāng)然你可以先用in操作檢查key是否在字典中再獲取,不過這種方式不符合Python之禪中說的:
Simpleisbetterthancomplex.
Flatisbetterthannested.
好的代碼應(yīng)該是簡(jiǎn)單易懂的,扁平的代碼結(jié)構(gòu)更加可讀。我們可以使用get方法來代替if…else
bad
d={'name':'python'}
if'name'ind:
print(d['hello'])
else:
print('default')
good
print(d.get("name","default"))
用setdefault為字典中不存在的key設(shè)置缺省值
data=[
("animal","bear"),
("animal","duck"),
("plant","cactus"),
("vehicle","speedboat"),
("vehicle","schoolbus")
]
在做分類統(tǒng)計(jì)時(shí),希望把同一類型的數(shù)據(jù)歸到字典中的某種類型中,比如上面代碼,把相同類型的事物用列表的形式重新組裝,得到新的字典
groups={}
>>>
{'plant':['cactus'],
'animal':['bear','duck'],
'vehicle':['speedboat','schoolbus']}
普通的方式就是先判斷key是否已經(jīng)存在,如果不存在則要先用列表對(duì)象進(jìn)行初始化,再執(zhí)行后續(xù)操作。而更好的方式就是使用字典中的setdefault方法。
bad
for(key,value)indata:
ifkeyingroups:
groups[key].append(value)
else:
groups[key]=[value]
good
groups={}
for(key,value)indata:
groups.setdefault(key,[]).append(value)
setdefault的作用是:
1、如果key存在于字典中,那么直接返回對(duì)應(yīng)的值,等效于get方法
2、如果key不存在字典中,則會(huì)用setdefault中的第二個(gè)參數(shù)作為該key的值,再返回該值。
用defaultdict初始化字典對(duì)象
如果不希望d[x]在x不存在時(shí)報(bào)錯(cuò),除了在獲取元素時(shí)使用get方法之外,另外一種方式是用collections模塊中的defaultdict,在初始化字典的時(shí)候指定一個(gè)函數(shù),其實(shí)defaultdit是dict的子類。
fromcollectionsimportdefaultdict
groups=defaultdict(list)
for(key,value)indata:
groups[key].append(value)
當(dāng)key不存在于字典中時(shí),list函數(shù)將被調(diào)用并返回一個(gè)空列表賦值給d[key],這樣一來,你就不用擔(dān)心調(diào)用d[k]會(huì)報(bào)錯(cuò)了。
用fromkeys將列表轉(zhuǎn)換成字典
keys={'a','e','i','o','u'}
value=[]
d=dict.fromkeys(keys,value)
print(d)
>>>
{'i':[],'u':[],'e':[],
'a':[],'o':[]}
用字典實(shí)現(xiàn)switch…case語句
Python中沒有switch…case語句,這個(gè)問題Python之父龜叔表示這個(gè)語法過去沒有,現(xiàn)在沒有,以后也不會(huì)有。因?yàn)镻ython簡(jiǎn)潔的語法完全可以用if…elif實(shí)現(xiàn)。如果有太多的分支判斷,還可以使用字典來代替。
ifarg==0:
return'zero'
elifarg==1:
return'one'
elifarg==2:
return"two"
else:
return"nothing"
good
data={
0:"zero",
1:"one",
2:"two",
}
data.get(arg,"nothing")
使用iteritems迭代字典中的元素
python提供了幾種方式迭代字典中的元素,第一種是使用items方法:
d={
0:"zero",
1:"one",
2:"two",
}
fork,vind.items():
print(k,v)
items方法返回的時(shí)(key,value)組成的列表對(duì)象,這種方式的弊端是迭代超大字典的時(shí)候,內(nèi)存瞬間會(huì)擴(kuò)大兩倍,因?yàn)榱斜韺?duì)象會(huì)一次性把所有元素加載到內(nèi)存,更好的方式是使用iteritems
fork,vind.iteritems():
print(k,v)
iteritems返回的是迭代器對(duì)象,迭代器對(duì)象具有惰性加載的特性,只有真正需要的時(shí)候才生成值,這種方式在迭代過程中不需要額外的內(nèi)存來裝載這些數(shù)據(jù)。注意Python3中,只有items方法了,它等價(jià)于Python2中的iteritems,而iteritems這個(gè)方法名被移除了。
使用字典推導(dǎo)式
推導(dǎo)式是個(gè)絕妙的東西,列表推導(dǎo)式一出,map、filter等函數(shù)黯然失色,自Python2.7以后的版本,此特性擴(kuò)展到了字典和集合身上,構(gòu)建字典對(duì)象無需調(diào)用dict方法。
bad
numbers=[1,2,3]
d=dict([(number,number*2)fornumberinnumbers])
good
numbers=[1,2,3]
d={number:number*2fornumberinnumbers}
以上內(nèi)容為大家介紹了Python之關(guān)于高效使用字典的清單,希望對(duì)大家有所幫助,如果想要了解更多Python相關(guān)知識(shí),請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。http://m.2667701.com/