字典(dict)對象是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)于獲取字典中的值，一種簡單的方式就是用d.[x]訪問該元素，但是這種情況在key不存在的情況下會報(bào)KeyError錯誤，當(dāng)然你可以先用in操作檢查key是否在字典中再獲取，不過這種方式不符合Python之禪中說的：

Simpleisbetterthancomplex.

Flatisbetterthannested.

好的代碼應(yīng)該是簡單易懂的，扁平的代碼結(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)存在，如果不存在則要先用列表對象進(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存在于字典中，那么直接返回對應(yīng)的值，等效于get方法

2、如果key不存在字典中，則會用setdefault中的第二個(gè)參數(shù)作為該key的值，再返回該值。

用defaultdict初始化字典對象

如果不希望d[x]在x不存在時(shí)報(bào)錯，除了在獲取元素時(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]會報(bào)錯了。

用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)在沒有，以后也不會有。因?yàn)镻ython簡潔的語法完全可以用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)組成的列表對象，這種方式的弊端是迭代超大字典的時(shí)候，內(nèi)存瞬間會擴(kuò)大兩倍，因?yàn)榱斜韺ο髸淮涡园阉性丶虞d到內(nèi)存，更好的方式是使用iteritems

fork,vind.iteritems():

print(k,v)

iteritems返回的是迭代器對象，迭代器對象具有惰性加載的特性，只有真正需要的時(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)建字典對象無需調(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)于高效使用字典的清單，希望對大家有所幫助，如果想要了解更多Python相關(guān)知識，請關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。http://gzshyw.cn/