字典(dict)對象是 Python 最常用的數據結構,社區曾有人開玩笑地說:"Python企圖用字典裝載整個世界",字典在Python中的重要性不言而喻,這里整理了幾個關于高效使用字典的清單,希望Python開發者可以在日常應用開發中合理利用,讓代碼更加 Pythonic。
1、用 in 關鍵字檢查 key 是否存在
Python之禪中有一條開發哲學是:
There should be one-- and preferably only one --obvious way to do it.
盡量找一種,最好是唯一種顯而易見的解決方案。Python2 中判斷某個 key 是否存在字典中可使用 has_key 方法,另外一種方式是使用 in 關鍵字。但是強烈推薦使用后者,因為 in 的處理速度更快,另外一個原因是 has_key 這個方法在 Python3 被移除了,要想同時兼容py2和py3兩個版本的代碼,用 in 是最好的選擇。
badd = {'name': 'python'}if d.has_key('name'): passgoodif 'name' in d: pass2、用 get 獲取字典中的值
關于獲取字典中的值,一種簡單的方式就是用d[x]訪問該元素,但是這種情況在 key 不存在的情況下會報 KeyError 錯誤,當然你可以先用 in 操作檢查 key 是否在字典中再獲取,不過這種方式不符合 Python 之禪中說的:
Simple is better than complex.
Flat is better than nested.
好的代碼應該是簡單易懂的,扁平的代碼結構更加可讀。我們可以使用 get 方法來代替 if ... else
badd = {'name': 'python'}if 'name' in d: print(d['hello'])else: print('default')goodprint(d.get("name", "default"))3、用 setdefault 為字典中不存在的 key 設置缺省值
data = [ ("animal", "bear"), ("animal", "duck"), ("plant", "cactus"), ("vehicle", "speed boat"), ("vehicle", "school bus") ]在做分類統計時,希望把同一類型的數據歸到字典中的某種類型中,比如上面代碼,把相同類型的事物用列表的形式重新組裝,得到新的字典
groups = {}>>> {'plant': ['cactus'], 'animal': ['bear', 'duck'], 'vehicle': ['speed boat', 'school bus']}普通的方式就是先判斷 key 是否已經存在,如果不存在則要先用列表對象進行初始化,再執行后續操作。而更好的方式就是使用字典中的 setdefault 方法。
badfor (key, value) in data: if key in groups: groups[key].append(value) else: groups[key] = [value]goodgroups = {}for (key, value) in data: groups.setdefault(key, []).append(value)setdefault 的作用是:
如果 key 存在于字典中,那么直接返回對應的值,等效于 get 方法
如果 key 不存在字典中,則會用 setdefault 中的第二個參數作為該 key 的值,再返回該值。
4、用 defaultdict 初始化字典對象
新聞熱點
疑難解答