我經(jīng)常遇到各種五花八門(mén)的關(guān)于Python字典的操作，今天就在這個(gè)文章中展示一些常見(jiàn)操作的優(yōu)化實(shí)現(xiàn)方式,，如果你都已經(jīng)掌握,，那么恭喜你，你已經(jīng)超越了90%以上的Python coder,。
通過(guò)序列生成字典
我們將下面的序列轉(zhuǎn)換為dict類型,。
lst = [('a', 1), ('b', 2), ('c', 3)]
普通的寫(xiě)法
for k, v in lst:
dic[k] = v
更pythonic的寫(xiě)法
利用字典推導(dǎo)式快速生成字典。
{k: v for k, v in lst}
key的默認(rèn)值
當(dāng)指定key不存在時(shí),，將value設(shè)置為 0,。
普通的寫(xiě)法
if key not in dct:
dct[key] = 0
pythonic的寫(xiě)法
dct[key] = dct.get(key, 0)
交換key與value
普通的寫(xiě)法
dic = {'Python': 1, 'Java': 2}
new_dic = {}
for k, v in dic.items():
new_dic[v] = k
更pythonic的寫(xiě)法
dic = {'Python': 1, 'Java': 2}
new_dic = {v: k for k, v in dic.items()}
序列修改和初始化
示例數(shù)據(jù)
lst = [('a', 1), ('b', 2), ('c', 3)]
dic = {'a': [0]}
如果我們需要根據(jù)lst來(lái)更新dic中的數(shù)據(jù)，當(dāng)key存在,，則將value添加到原序列末尾,，否則初始化value并用序列保存。
普通的寫(xiě)法
for key, value in lst:
if key in dic:
dic[key].append(value)
else:
dic[key] = [value]
更pythonic的寫(xiě)法
for (key, value) in lst:
group = dic.setdefault(key, [])
group.append(value)
# dic：{'a': [0, 1], 'b': [2], 'c': [3]}
setdefault(key, default)會(huì)先判斷key是否存在,，存在則返回dct[key] , 不存在則把dct[key]設(shè)為 [] 并返回,。
key，items的集合運(yùn)算
如果我們現(xiàn)在需要獲取兩個(gè)字典的key相交的部分的映射信息,。
普通的寫(xiě)法
dic1 = {'Python': 1, 'Java': 2, 'C': 3}
dic2 = {'Python': 3, 'Java': 2, 'C++': 1}
new_dic = {}
for k, v in dic1.items():
if k in dic2.keys():
new_dic[k] = v
print(new_dic)
# {'Python': 1, 'Java': 2}
更pythonic的寫(xiě)法
dic1 = {'Python': 1, 'Java': 2, 'C': 3}
dic2 = {'Python': 3, 'Java': 2, 'C++': 1}
print({k: dic1[k] for k in dic1.keys() & dic2.keys()})
# {'Python': 1, 'Java': 2}
這里的dic1.keys() & dic2.keys()用到的就是 keys()進(jìn)行集合運(yùn)算,，items()同樣可以進(jìn)行集合運(yùn)算。
如果現(xiàn)在我們要獲取兩個(gè)字典中 key,，value 完全相同的部分,。
dic1 = {'Python': 1, 'Java': 2, 'C': 3}
dic2 = {'Python': 3, 'Java': 2, 'C++': 1}
print(dic1.items() & dic2.items())
# {('Java', 2)}
靈活運(yùn)用 keys，items() 集合運(yùn)算的特性,，可以快速提取我們想要的內(nèi)容,。
按key或value對(duì)字典排序
使用sorted()函數(shù)快速實(shí)現(xiàn)對(duì)key或value的排序。
dic = {'a': 2, 'b': 1, 'c': 3, 'd': 0}
lst1 = sorted(dic.items(), key=lambda x: x[0], reverse=False)
# [('a', 2), ('b', 1), ('c', 3), ('d', 0)]
lst2 = sorted(dic.items(), key=lambda x: x[1], reverse=False)
# [('d', 0), ('b', 1), ('a', 2), ('c', 3)]
print('按照鍵降序：', {key: value for key, value in lst1})
print('按照值降序：', {key: value for key, value in lst2})
# 按照鍵降序： {'a': 2, 'b': 1, 'c': 3, 'd': 0}
# 按照值降序： {'d': 0, 'b': 1, 'a': 2, 'c': 3}
多個(gè)字典排序
如果一個(gè)序列中包含多個(gè)字典,，現(xiàn)在要根據(jù)條件對(duì)這些字典繼續(xù)排序,。同樣可以使用sorted()函數(shù)來(lái)實(shí)現(xiàn)。
dict_list = [
{'letter': 'B', 'number': '2'},
{'letter': 'A', 'number': '3'},
{'letter': 'B', 'number': '1'}
]
# 按 letter 排序
print(sorted(dict_list,
key=lambda dic: dic['letter']))
# 按 letter, number 排序
print(sorted(dict_list,
key=lambda dic: (dic['letter'], dic['number'])))
# [{'letter': 'A', 'number': '3'}, {'letter': 'B', 'number': '2'}, {'letter': 'B', 'number': '1'}]
# [{'letter': 'A', 'number': '3'}, {'letter': 'B', 'number': '1'}, {'letter': 'B', 'number': '2'}]
當(dāng)然，如果你知道itemgetter()的話，上面的代碼就可以改變一下,，執(zhí)行速度會(huì)更快,。
from operator import itemgetter
print(sorted(dict_list,
key=itemgetter('letter')))
print(sorted(dict_list,
key=itemgetter('letter', 'number')))
itemgetter()獲取的不是值，而是定義了一個(gè)函數(shù),，通過(guò)該函數(shù)作用到目標(biāo)對(duì)象上,。
