Python字典：竟還有我不會的高階玩法？

2022-02-22 11:50:16

我經常遇到各種五花八門的關于Python字典的操作，今天就在這個文章中展示一些常見操作的優化實現方式，如果你都已經掌握，那么恭喜你，你已經超越了90%以上的Python coder。

通過序列生成字典

我們將下面的序列轉換為dict類型。

lst = [('a', 1), ('b', 2), ('c', 3)]

普通的寫法

for k, v in lst:
    dic[k] = v

更pythonic的寫法

利用字典推導式快速生成字典。

{k: v for k, v in lst}

key的默認值

當指定key不存在時，將value設置為 0。

普通的寫法

if key not in dct:
    dct[key] = 0

pythonic的寫法

dct[key] = dct.get(key, 0)

交換key與value

普通的寫法

dic = {'Python': 1, 'Java': 2}
new_dic = {}
for k, v in dic.items():
    new_dic[v] = k

更pythonic的寫法

dic = {'Python': 1, 'Java': 2}
new_dic = {v: k for k, v in dic.items()}

序列修改和初始化

示例數據

lst = [('a', 1), ('b', 2), ('c', 3)]
dic = {'a': [0]}

如果我們需要根據lst來更新dic中的數據，當key存在，則將value添加到原序列末尾，否則初始化value并用序列保存。

普通的寫法

for key, value in lst:
    if key in dic:
        dic[key].append(value)
    else:
        dic[key] = [value]

更pythonic的寫法

for (key, value) in lst:
    group = dic.setdefault(key, [])
    group.append(value)
# dic：{'a': [0, 1], 'b': [2], 'c': [3]}

setdefault(key, default)會先判斷key是否存在，存在則返回dct[key] , 不存在則把dct[key]設為 [] 并返回。

key，items的集合運算

如果我們現在需要獲取兩個字典的key相交的部分的映射信息。

普通的寫法

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的寫法

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()進行集合運算，items()同樣可以進行集合運算。

如果現在我們要獲取兩個字典中 key，value 完全相同的部分

dic1 = {'Python': 1, 'Java': 2, 'C': 3}
dic2 = {'Python': 3, 'Java': 2, 'C++': 1}
print(dic1.items() & dic2.items())
# {('Java', 2)}

靈活運用 keys，items() 集合運算的特性，可以快速提取我們想要的內容。

按key或value對字典排序

使用sorted()函數快速實現對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}

多個字典排序

如果一個序列中包含多個字典，現在要根據條件對這些字典繼續排序。同樣可以使用sorted()函數來實現。

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'}]

當然，如果你知道itemgetter()的話，上面的代碼就可以改變一下，執行速度會更快。

from operator import itemgetter
print(sorted(dict_list
             key=itemgetter('letter')))
print(sorted(dict_list,
             key=itemgetter('letter', 'number')))

itemgetter()獲取的不是值，而是定義了一個函數，通過該函數作用到目標對象上。

責任編輯：龐桂玉來源：菜鳥學Python

Python 字典代碼