【Python基础语法】defaultdict

Python的defaultdict是collections模块中提供的一种特殊的字典类型，

它与普通的字典（dict）有着相似的功能，但有一个关键的不同点：当访问一个不存在的键时，defaultdict不会抛出KeyError异常，而是会自动为该键创建一个默认值。

这种特性使得在处理需要初始化新键的情况时更加方便。

创建defaultdict
要使用defaultdict，首先需要从collections模块导入它：

from collections import defaultdict

然后可以创建一个defaultdict实例，传入一个可调用对象作为default_factory参数。
这个可调用对象决定了当访问不存在的键时应该返回的默认值类型。
例如：

• 使用int作为default_factory将返回整数0。
• 使用list作为default_factory将返回空列表[]。
• 使用set作为default_factory将返回空集合set()。
• 使用自定义函数作为default_factory将根据该函数的返回值来确定默认值。

示例1

from collections import defaultdict

bag = ['apple', 'orange', 'cherry', 'apple','apple', 'cherry', 'blueberry']
count = defaultdict(int)  # 使用int作为default_factory
print(f"\n > > > > > > count:\n {count}")
print(f"\n > > > > > > type(count):\n {type(count)}")

for fruit in bag:
    count[fruit] += 1
print(f"\n > > > > > > count:\n {count}")

输出：

 > > > > > > count:
 defaultdict(<class 'int'>, {})

 > > > > > > type(count):
 <class 'collections.defaultdict'>

 > > > > > > count:
 defaultdict(<class 'int'>, {'apple': 3, 'orange': 1, 'cherry': 2, 'blueberry': 1})

在这个例子中，对一个不存在的键进行操作时，defaultdict会自动为该键分配一个默认值0，并允许直接对其进行递增操作。

示例2

from collections import defaultdict

names = [('group1', 'Alice'), ('group2', 'Bob'), ('group1', 'Charlie')]
grouped_names = defaultdict(list)
print(f"\n > > > > > > grouped_names:\n {grouped_names}")
print(f"\n > > > > > > type(grouped_names):\n {type(grouped_names)}")

for group, name in names:
    grouped_names[group].append(name)

print(f"\n > > > > > > grouped_names:\n {grouped_names}")

输出：

 > > > > > > grouped_names:
 defaultdict(<class 'list'>, {})

 > > > > > > type(grouped_names):
 <class 'collections.defaultdict'>

 > > > > > > grouped_names:
 defaultdict(<class 'list'>, {'group1': ['Alice', 'Charlie'], 'group2': ['Bob']})

在这个例子中，根据第一个元素对名字进行了分组。如果使用普通字典，则需要额外的逻辑来检查和初始化每个新组。