首頁  >  文章  >  後端開發  >  如何在Python中使用defaultdict設定預設值?

如何在Python中使用defaultdict設定預設值?

WBOY
WBOY轉載
2023-04-26 17:07:091300瀏覽

前言

在字典中尋找某一個值的時候,若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 &#39;int&#39;>, {0: 0})
print(dict2) # defaultdict(<class &#39;set&#39;>, {0: set()})
print(dict3) # defaultdict(<class &#39;str&#39;>, {0: &#39;&#39;})
print(dict4) # defaultdict(<class &#39;list&#39;>, {1: 666, 2: 888, 0: []})

setdefault()和defaultdict()的區別

dict.setdefault()方法有兩個參數,第一個是參數,第二個是預設值

lst = [&#39;A&#39;, &#39;B&#39;, &#39;C&#39;, &#39;D&#39;, &#39;e&#39;]
dic = {}
 
for i in lst:
    dic.setdefault(i, 0)
    dic[i] += 1
print(dic)# {&#39;A&#39;: 1, &#39;B&#39;: 1, &#39;C&#39;: 1, &#39;D&#39;: 1, &#39;e&#39;: 1}

setdefault()

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) # {&#39;A&#39;: &#39;a&#39;, &#39;B&#39;: &#39;b&#39;}
 
dic2 = {"C": "c", "D": "d"}
y = dic2.setdefault("E", &#39;e&#39;)
print(y) # e
print(dic2) # {&#39;C&#39;: &#39;c&#39;, &#39;D&#39;: &#39;d&#39;, &#39;E&#39;: &#39;e&#39;}

defaultdict()

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設定預設值?

#

以上是如何在Python中使用defaultdict設定預設值?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

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