在字典中尋找某一個值的時候,若key不存在就會回傳一個keyerror錯誤而不是一個預設值,如果想要回傳一個預設值可以使用
defaultdict
函數。 defaultdict是python內建dict類的一個字類,功能與dict相同,但它帶有一個預設的值,若key值不存在時返回一個預設的值。
dict =defaultdict( factory_function)
defaultdict
可以接受一個工廠函數作為參數,其中factory_function
可以是list、set、str等,作用是當ket不存在的時候,回傳一個工廠函數的預設值,例如list對應[ ],str對應的是空字串,set對應set( ),int對應0。
正常使用dict如下:
l=['A','B','C','D'] di={} for i in l: print(i) # 'A di[i]+=1 # KeyError: 'A' print(di)
使用defaultdict()
from collections import defaultdict dict1 = defaultdict(int) dict2 = defaultdict(set) dict3 = defaultdict(str) dict4 = defaultdict(list) dict4[1] = 666 dict4[2] = 888 print(dict1[0]) # 0 print(dict2[0]) # set() print(dict3[0]) # print(dict4[0]) # [] print(dict4[1]) # 666 print("*" * 30) print(dict1) # defaultdict(<class 'int'>, {0: 0}) print(dict2) # defaultdict(<class 'set'>, {0: set()}) print(dict3) # defaultdict(<class 'str'>, {0: ''}) print(dict4) # defaultdict(<class 'list'>, {1: 666, 2: 888, 0: []})
dict.setdefault()方法有兩個參數,第一個是參數,第二個是預設值
lst = ['A', 'B', 'C', 'D', 'e'] dic = {} for i in lst: dic.setdefault(i, 0) dic[i] += 1 print(dic)# {'A': 1, 'B': 1, 'C': 1, 'D': 1, 'e': 1}
setdefault()是字典的一個實例方法,接收兩個參數,用法和字典的get()方法相似,但是比get()方法更強大。都為字典的key設定一個預設值。
二者的差異是:get 方法設定的預設值不會改變原始字典,而setdefault設定的預設值會改變原字典的值。
注意:使用dict[key]=value時,若key不存在則報錯;使用dict.get(key)時,若key不存在則會傳回預設值。
dic1 = {"A": "a", "B": "b"} x = dic1.get("E", "e") print(x) # e print(dic1) # {'A': 'a', 'B': 'b'} dic2 = {"C": "c", "D": "d"} y = dic2.setdefault("E", 'e') print(y) # e print(dic2) # {'C': 'c', 'D': 'd', 'E': 'e'}
defaultdict是屬於collections 模組下的一個工廠函數,用於建構字典對象,接收一個函數(可呼叫)物件作為參數。參數回傳的型別是什麼,key對應value就是什麼型別。
from collections import defaultdict lst = [("A", "1"), ("B", "1"), ("A", "2"), ("B", "2"), ("A", "3"), ("B", "3")] dic = defaultdict(list) print(dic) for key, value in lst: dic[key].append(value) print(dic) print(type(dic)) for key, value in dic.items(): print(key, value)
以上是如何在Python中使用defaultdict設定預設值?的詳細內容。更多資訊請關注PHP中文網其他相關文章!