Python defaultdict 的强大之处:告别繁琐的字典键检查: Effective Python 第17条

在 Python 编程中,字典是一种非常灵活和强大的数据结构,允许我们存储键值对。然而,当需要处理不存在的键时,我们可能会遇到一些挑战。defaultdict 是 Python collections 模块中的一个类,它能够自动为缺失的键分配默认值,从而简化了字典操作。本文将通过具体的例子来展示 defaultdict 的用法和优势。

问题背景

假设我们有一个需求,要记录一个人去过的国家和城市。我们可以使用一个字典来实现,其中键是国家,值是一个集合,集合中包含该国家访问过的城市。

使用 setdefault 的方式

一种常见的做法是使用 setdefault 方法。该方法会在键不存在时,为该键设置一个默认值,然后返回该键对应的值。

python 复制代码
visits = {
    'Mexico': {'Tulum', 'Puerto Vallarta'},
    'Japan': {'Hakone'},
}

visits.setdefault('France', set()).add('Arles')
if (japan := visits.get('Japan')) is None:
    visits['Japan'] = japan = set()
japan.add('Kyoto')

这种方法虽然可行,但代码不够直观,且每次调用 add 方法时,即使键已存在,也会创建一个新的 set 实例,这显然是不高效的。

使用 defaultdict 优化

Python 的 collections 模块提供了一个名为 defaultdict 的类,它能够解决这个问题。defaultdict 允许我们在键不存在时,自动为该键分配一个默认的值。这样我们就不需要检查键是否存在,也不需要手动设置默认值。

下面是如何使用 defaultdict 来重构前面的例子:

python 复制代码
from collections import defaultdict

class Visits:
    def __init__(self):
        # 使用 defaultdict 自动创建缺失的集合
        self.data = defaultdict(set)

    def add(self, country, city):
        # 直接添加城市到国家的集合中
        self.data[country].add(city)

visits = Visits()
visits.add('England', 'Bath')
visits.add('England', 'London')
visits.add('France', 'Paris')
visits.add('France', 'Lyon')
visits.add('Japan', 'Tokyo')
visits.add('Japan', 'Kyoto')
visits.add('Japan', 'Osaka')
visits.add('Italy', 'Rome')
visits.add('Italy', 'Venice')

print(visits.data)

打印结果:

复制代码
defaultdict(<class 'set'>, {
    'England': {'Bath', 'London'},
    'France': {'Paris', 'Lyon'},
    'Japan': {'Tokyo', 'Kyoto', 'Osaka'},
    'Italy': {'Rome', 'Venice'}
})

实际应用中的 defaultdict 例子

让我们通过几个具体的例子来看看 defaultdict 如何在真实世界的应用中发挥作用。

例1:单词计数

python 复制代码
class WordCounter:
    def __init__(self):
        self.word_counts = defaultdict(int)

    def process_text(self, text):
        words = text.split()
        for word in words:
            self.word_counts[word] += 1

word_counter = WordCounter()
word_counter.process_text("Hello world! Hello everyone.")
print(word_counter.word_counts)

打印结果:

复制代码
defaultdict(<class 'int'>, {'Hello': 2, 'world!': 1, 'everyone.': 1})

例2:用户活动跟踪

python 复制代码
class UserActivity:
    def __init__(self):
        self.activity = defaultdict(set)

    def log_activity(self, user, page):
        self.activity[user].add(page)

user_activity = UserActivity()
user_activity.log_activity('user1', '/home')
user_activity.log_activity('user1', '/about')
user_activity.log_activity('user2', '/contact')

print(user_activity.activity)

打印结果:

复制代码
defaultdict(<class 'set'>, {'user1': {'/home', '/about'}, 'user2': {'/contact'}})

例3:购物车

python 复制代码
class ShoppingCart:
    def __init__(self):
        self.cart = defaultdict(int)

    def add_item(self, user, item):
        self.cart[(user, item)] += 1

shopping_cart = ShoppingCart()
shopping_cart.add_item('user1', 'apple')
shopping_cart.add_item('user1', 'banana')
shopping_cart.add_item('user2', 'apple')

print(shopping_cart.cart)

打印结果:

复制代码
defaultdict(<class 'int'>, {('user1', 'apple'): 1, ('user1', 'banana'): 1, ('user2', 'apple'): 1})

结论

通过上述例子,我们可以看到 defaultdict 在处理需要动态创建字典键的情况时的便利性和效率。它简化了代码,减少了错误,并且提高了性能。在处理动态数据结构时,defaultdict 是一个值得推荐的工具。通过使用 defaultdict,我们可以写出更简洁、更易读、更高效的代码。

相关推荐
etsuyou1 小时前
js前端this指向规则
开发语言·前端·javascript
shizhenshide1 小时前
为什么有时候 reCAPTCHA 通过率偏低,常见原因有哪些
开发语言·php·验证码·captcha·recaptcha·ezcaptcha
lichong9511 小时前
Git 检出到HEAD 再修改提交commit 会消失解决方案
java·前端·git·python·github·大前端·大前端++
Tiny番茄1 小时前
31.下一个排列
数据结构·python·算法·leetcode
mit6.8241 小时前
[Agent可视化] 配置系统 | 实现AI模型切换 | 热重载机制 | fsnotify库(go)
开发语言·人工智能·golang
友友马1 小时前
『 QT 』QT控件属性全解析 (一)
开发语言·前端·qt
小白学大数据2 小时前
实战:Python爬虫如何模拟登录与维持会话状态
开发语言·爬虫·python
一念&2 小时前
每日一个C语言知识:C 结构体
c语言·开发语言
FriendshipT2 小时前
目标检测:使用自己的数据集微调DEIMv2进行物体检测
人工智能·pytorch·python·目标检测·计算机视觉
平谷一勺2 小时前
数据清洗-缺失值的处理
python·数据分析