快捷導(dǎo)航

詳解python中的defaultdict?默認(rèn)值

更新時(shí)間：2023年04月15日 10:13:47 作者：@@老胡

這篇文章主要介紹了defaultdict?默認(rèn)值,defaultdict是python內(nèi)建dict類的一個(gè)字類，功能與dict相同，但它帶有一個(gè)默認(rèn)的值，若key值不存在時(shí)返回一個(gè)默認(rèn)的值，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，需要的朋友可以參考下

前言

在字典中查找某一個(gè)值的時(shí)候，若key不存在就會(huì)返回一個(gè)keyerror錯(cuò)誤而不是一個(gè)默認(rèn)值，如果想要返回一個(gè)默認(rèn)值可以使用defaultdict函數(shù)。 defaultdict是python內(nèi)建dict類的一個(gè)字類，功能與dict相同，但它帶有一個(gè)默認(rèn)的值，若key值不存在時(shí)返回一個(gè)默認(rèn)的值。

語法

dict =defaultdict( factory_function)

defaultdict可以接受一個(gè)工廠函數(shù)作為參數(shù)，其中factory_function可以是list、set、str等，作用是當(dāng)ket不存在的時(shí)候，返回一個(gè)工廠函數(shù)的默認(rèn)值，比如list對(duì)應(yīng)[ ]，str對(duì)應(yīng)的是空字符串，set對(duì)應(yīng)set( )，int對(duì)應(yīng)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: []})

setdefault()和defaultdict()的區(qū)別

dict.setdefault()方法有兩個(gè)參數(shù)，第一個(gè)是參數(shù)，第二個(gè)是默認(rèn)值

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()

setdefault()是字典的一個(gè)實(shí)例方法，接收兩個(gè)參數(shù)，用法和字典的get()方法相似，但是比get()方法更加強(qiáng)大。都為字典的key設(shè)置一個(gè)默認(rèn)值。

二者的區(qū)別是：get 方法設(shè)置的默認(rèn)值不會(huì)改變原字典，而setdefault設(shè)置的默認(rèn)值會(huì)改變原字典的值。

注意：使用dict[key]=value時(shí)，若key不存在則報(bào)錯(cuò)；使用dict.get(key)時(shí)，若key不存在則會(huì)返回一個(gè)默認(rèn)值。

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()

defaultdict是屬于collections 模塊下的一個(gè)工廠函數(shù)，用于構(gòu)建字典對(duì)象，接收一個(gè)函數(shù)（可調(diào)用）對(duì)象為作為參數(shù)。參數(shù)返回的類型是什么，key對(duì)應(yīng)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)

在這里插入圖片描述