在Python中需要初始化字典的时候我一般会调用collections
中的defaultdict
来实现,这样可以减少代码行数,并且避免更复杂的逻辑判断,比如:
a_list = ['a', 'b', 'c', 'a', 'd']
# 常规方法
count_dict = {}
for a in a_list:
if a in count_dict:
count_dict[a] += 1
else:
# 如果dict中不存在该key,则需要赋值来初始化
count_dict[a] = 1
# 使用defaultdict进行初始化只需要4行代码,且不需要if...else进行逻辑判断
import collections
count_dict = collections.defaultdict(int)
for a in a_list:
count_dict[a] += 1