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']})

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

到此这篇关于Python基础语法中defaultdict的使用小结的文章就介绍到这了,更多相关Python defaultdict内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

最新评论