首頁  >  文章  >  後端開發  >  Python字典:竟還有我不會的高階玩法?

Python字典:竟還有我不會的高階玩法?

WBOY
WBOY轉載
2023-04-11 23:58:081564瀏覽

Python字典:竟還有我不會的高階玩法?


透過序列產生字典

我們將下面的序列轉換為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字典:竟還有我不會的高階玩法?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:51cto.com。如有侵權,請聯絡admin@php.cn刪除