首页 > 文章列表 > Python中如何使用defaultdict实现默认值操作

Python中如何使用defaultdict实现默认值操作

Python defaultdict
255 2023-04-29

python中的defaultdict默认值怎么应用

前言

在字典中查找某一个值的时候,若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: []})

setdefault()和defaultdict()的区别

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

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

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)