Table of contents
defaultdict会自动给key赋予初始value是0,不用处理key不在dict()的异常情况。
举个例子:
使用 dict()
当使用普通字典时,如果尝试访问或修改一个不存在的键,你需要手动处理这种情况。例如,在统计字符出现次数的例子中,使用普通字典需要额外的代码来检查键是否存在:
text = "hello world"
char_count = {}
for char in text:
if char in char_count:
char_count[char] += 1
else:
char_count[char] = 1
print(char_count)
在这个例子中,你必须使用if-else
语句来检查每个字符是否已经在字典中。如果字符不在字典中,你需要先初始化它的计数为1,然后才能进行累加操作。
使用 defaultdict
相比之下,defaultdict
自动处理不存在的键。如果你尝试访问或修改一个不存在的键,defaultdict
会使用提供的工厂函数来为该键创建一个默认值,然后返回这个值。这样可以减少代码量并提高可读性:
from collections import defaultdict
text = "hello world"
char_count = defaultdict(int) # 默认值是0
print(char_count)
for char in text:
char_count[char] += 1
print(char_count)
print(char_count)
在这个例子中,你不需要检查字符是否已经存在于字典中。如果字符不存在,defaultdict
会自动创建一个值为0的条目,然后进行累加操作。这使得代码更加简洁和直观。
上述的代码会输出hello world各个字母出现的次数,结果是:
defaultdict(<class 'int'>, {})
defaultdict(<class 'int'>, {'h': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 2})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 2, 'o': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 2, 'o': 1, ' ': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 2, 'o': 1, ' ': 1, 'w': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 2, 'o': 2, ' ': 1, 'w': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 2, 'o': 2, ' ': 1, 'w': 1, 'r': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 3, 'o': 2, ' ': 1, 'w': 1, 'r': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 3, 'o': 2, ' ': 1, 'w': 1, 'r': 1, 'd': 1})
defaultdict(<class 'int'>, {'h': 1, 'e': 1, 'l': 3, 'o': 2, ' ': 1, 'w': 1, 'r': 1, 'd': 1})