Python技巧：getOrDefault方法的妙用

小蓝 • 2024-12-12 13:05 • 编程

一、什么是getOrDefault方法

在Python中，字典是一种非常有用的数据结构。通常，当我们需要查询字典中某个键对应的值时，我们会使用字典中的get方法。例如：

d = {'a': 1, 'b': 2, 'c': 3}
print(d.get('a', 0)) # 输出1
print(d.get('d', 0)) # 输出0

在这个例子中，当我们使用get方法查询字典中’a’对应的值时，返回了1。而当我们查询字典中不存在的键’d’对应的值时，由于使用了get方法的默认参数，返回了0。

然而，有时候我们希望设置某个键的默认值，即在查询时若该键不存在，则返回我们预设的默认值。这时我们可以使用Python 3.7中新增的getOrDefault方法。

class DefaultDict(dict):
    def __init__(self, def_val):
        self.def_val = def_val
    def getOrDefault(self, key):
        return self.get(key, self.def_val)
 
d = DefaultDict(0)
d['a'] = 1
print(d.getOrDefault('a')) # 输出1
print(d.getOrDefault('b')) # 输出0

在这个例子中，我们创建了一个继承自Python原生字典类的DefaultDict类，并定义了一个getOrDefault方法，其中的self.def_val是我们预设的默认值。当我们使用getOrDefault方法时，若键存在则返回对应的值，若键不存在则返回预设的默认值0。

二、getOrDefault方法的优点

那么，为什么我们需要getOrDefault方法呢？和get方法相比，getOrDefault方法有以下几个优点：

1. 简洁易懂

相对于使用get方法时需要传入许多参数的繁琐操作，getOrDefault方法更加简洁易懂。在使用自定义字典的情况下，更容易理解预设的默认值。

2. 不会改变原字典中的值

使用Python原生字典的get方法时，会直接返回字典中某个键对应的值，而在返回此值的同时还会将原字典的内容改变（增加了None）。相对而言，使用getOrDefault方法时只返回预设的值，不会改变原字典中的内容。

三、getOrDefault方法的应用

getOrDefault方法在实际工作中有许多应用场景。例如在机器学习领域中，经常需要处理大量的文本数据，需要进行特征提取，然后将每个文本的特征表示成一个向量。这时使用字典来记录每个单词的出现次数十分常见。

class WordCount:
    def __init__(self):
        self.word_count = DefaultDict(0)
    def add(self, sentence):
        for word in sentence.split():
            self.word_count[word] += 1
    def count(self, word):
        return self.word_count.getOrDefault(word)
    
wc = WordCount()
wc.add("I love Python!")
wc.add("Python is easy to learn.")
print(wc.count('Python')) # 输出2
print(wc.count('Java')) # 输出0

在这个例子中，我们首先定义了一个WordCount类，其中的self.word_count是我们记录单词出现次数的字典。在add函数中，我们将每个句子中出现的单词计数，并将结果保存在word_count字典中。在count函数中，我们使用getOrDefault方法获取某个单词的出现次数，若该单词不存在则返回预设的默认值0。

四、总结

本文介绍了Python 3.7中新增的getOrDefault方法，并列举了其与get方法相比的优点，讨论了其在实际应用中的用法，希望可以帮助读者更好地理解Python中字典的相关操作。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/244940.html