本文始发于个人公众号:TechFlow,原创不易,求个关注
今天为大家介绍Python当中一个很好用也是很基础的工具库,叫做collections。
collection在英文当中有容器的意思,所以顾名思义,这是一个容器的集合。这个库当中的容器很多,有一些不是很常用,本篇文章选择了其中最常用的几个,一起介绍给大家。
defaultdict
defaultdict可以说是这个库当中使用最简单的一个,并且它的定义也很简单,我们从名称基本上就能看得出来。它解决的是我们使用dict当中最常见的问题,就是key为空的情况。
在正常情况下,我们在dict中获取元素的时候,都需要考虑key为空的情况。如果不考虑这点,那么当我们获取了一个不存在的key,会导致系统抛出异常。我们当然可以在每次get之前写一个if判断,但是这很麻烦,比如:
if key in dict:
return dict[key]
else:
return None
当然,这是最笨的方法,dict当中为我们提供了带默认值的get方法。比如,我们可以写成:
return dict.get(key, None)
这样,当key不在dict当中存在的时候,会自动返回我们设置的默认值。这个省去了很多麻烦的判断,但是在一些特殊情况下仍然存在一点问题。举个例子,比如当key存在重复,我们希望将key相同的value存进一个list当中,而不是只保留一个。这种情况下写成代码就会比较复杂:
data = [(1, 3), (2, 1), (1, 4), (2, 5), (3, 7)]
d = {
}
for k, v in data:
if k