python中defaultdict用法详解

defaultdict底层代码：

        在字典中查找某个值时，若key不存在时则会返回一个KeyError错误而不是一个默认值，这时候可以使用defaultdict函数。

注意：使用dict[key]时，若key不存在则报错；使用dict.get(key)时，若key不存在则会返回一个默认值。

注意：key不存在dict当中时，dict[key] = value是赋值，不会报错；key不存在dict当中，以dict[key]方式获取值，会报错

defaultdict接受一个工厂函数作为参数，如下来构造：

dict =defaultdict( factory_function)

        factory_function可以是list、set、str等等，作用是当key不存在时，返回的是工厂函数的默认值，比如list对应[ ]，str对应的是空字符串，set对应set( )，int对应0。

        defaultdict是python内建dict类的一个字类，功能与dict相同，但它带有一个默认的值，若key值不存在时返回一个默认的值。

示例代码：

lst = ['A', 'B', 'C', 'D', 'e']
dic = {}

for i in lst:
    dic[i] += 1
print(dic)

若访问字典中的key值不存在时会报KeyError错误，这时候就可以使用defaultdict类来避免这种错误。

 dict.setdefault()方法：有两个参数，第一个是参数，第二个是默认值。

示例代码：

lst = ['A', 'B', 'C', 'D', 'e']
dic = {}

for i in lst:
    dic.setdefault(i, 0)
    dic[i] += 1
print(dic)

运行结果： 

示例代码：

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])
print(dict2[0])
print(dict3[0])
print(dict4[0])
print(dict4[1])

print("*" * 100)
print(dict1)
print(dict2)
print(dict3)
print(dict4)

运行效果：

setdefault()和defaultdict()的区别：

setdefault()

        setdefault()是字典的一个实例方法，接收两个参数，用法和字典的get()方法相似，但是比get()方法更加强大。都为字典的key设置一个默认值。

二者的区别是：get 方法设置的默认值不会改变原字典， 而setdefault设置的默认值会改变原字典的值。

示例代码：

dic1 = {"A": "a", "B": "b"}
x = dic1.get("E", "e")
print(x)
print(dic1)

dic2 = {"C": "c", "D": "d"}
y = dic2.setdefault("E", 'e')
print(y)
print(dic2)

dic3 = {"C": "c", "D": "d"}
z = dic3.setdefault("C", 'e')
print(z)
print(dic3)

运行结果： 

 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)
    print(type(key), type(value))

运行结果：