首页 > 代码库 > 简介Python的collections模块中defaultdict类型
简介Python的collections模块中defaultdict类型
这里我们来简介Python的collections模块中defaultdict类型的用法,与内置的字典类最大的不同在于初始化上,一起来看一下:
defaultdict 主要用来需要对 value 做初始化的情形。对于字典来说,key 必须是 hashable,immutable,unique 的数据,而 value 可以是任意的数据类型。如果 value 是 list,dict 等数据类型,在使用之前必须初始化为空,有些情况需要把 value 初始化为特殊值,比如 0 或者 ‘‘。
from collections import defaultdict person_by_age = defaultdict(list) for person in persons: d[person.age].append(person.name)
defaultdict 和 dict 的使用方法一样,只有在初始化的时候必须传入一个 callable 的对象 x,当访问某个还不存在的 key 时,会把 value 自动设置成 x()。比如上例中,当第一次访问某个年龄的人 d[person.age] 就会变成 list(), 也就是 []。
当然也可以使用自己定义的 callable 对象,比如:
d = defaultdict(lambda: 0) d["hello"] += 1 # 1 d["a"] # 0
defaultdict 要比 dict.set_default 效率更高,使用起来也更直观和方便。
标准字典包括一个方法setdefault()来获取一个值,如果值不存在则建立一个默认值。defaultdict初始化容器是会让调用者提前指定默认值。
import collections def default_factory(): return ‘default value‘ d = collections.defaultdict(default_factory, foo = ‘bar‘) print ‘d:‘, d print ‘foo =>‘, d[‘foo‘] print ‘var =>‘, d[‘bar‘]
只要所有键都有相同的默认值,就可以使用这个方法。
上面的结果是:
d: defaultdict(<function default_factory at 0x0201FAB0>, {‘foo‘: ‘bar‘}) foo => bar var => default value
其他实例
1 #dic1 = {}
2 #dic1["k1"].append("kkkkk") #正常情况会报错
3 #print(dic1)
4
5 import collections
6 dic = collections.defaultdict(list) #指定字典的值为列表
7 dic["k1"].append("kkkkk")
8 print(dic)
9
10 #执行结果:
11 defaultdict(<class ‘list‘>, {‘k1‘: [‘kkkkk‘]})
1 #有如下值集合,不用默认字典执行代码如下
2 #[11,22,33,44,55,66,77,88,99,90]
3 #将所有大于66的值保存在字典的第一个key中,
4 #将小于等于66的值保存在第二个key的值(列表)中。
5 #即:{"k1":大于66,"k2":小于等于66}
6
7 li = [11,22,33,44,55,66,77,88,99,90]
8 dic = {}
9
10 for i in li:
11 if i>66:
12 if "k1" in dic.keys():
13 dic["k1"].append(i)
14 else:
15 dic["k1"] = [i,]
16 else:
17 if "k2" in dic.keys():
18 dic["k2"].append(i)
19 else:
20 dic["k2"] = [i,]
21 print(dic)
22
23 #执行结果:
24 {‘k2‘: [11, 22, 33, 44, 55, 66], ‘k1‘: [77, 88, 99, 90]}
1 #用默认字典精简代码
2 from collections import defaultdict #这样写,下面就不用写collections.defaultdict了
3 values = [11, 22, 33,44,55,66,77,88,99,90]
4 my_dict = defaultdict(list)
5
6 for value in values:
7 if value>66:
8 my_dict[‘k1‘].append(value)
9 else:
10 my_dict[‘k2‘].append(value)
11 print(my_dict)
12
13 #执行结果:
14 defaultdict(<class ‘list‘>, {‘k2‘: [11, 22, 33, 44, 55, 66], ‘k1‘: [77, 88, 99, 90]})
简介Python的collections模块中defaultdict类型
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。