处理 Python 字典中缺失的键

---

在 Python 中，有一种容器称为字典。在字典中，我们可以将键映射到其值。使用字典，可以以常数时间访问值。但是，当给定的键不存在时，可能会发生一些错误。

在本节中，我们将了解如何处理此类错误。如果我们尝试访问缺失的键，它可能会返回如下错误。

示例代码

实时演示

country_dict = {'India' : 'IN', 'Australia' : 'AU', 'Brazil' : 'BR'}
print(country_dict['Australia'])
print(country_dict['Canada']) # This will return error

输出

AU
---------------------------------------------------------------------------
KeyErrorTraceback (most recent call last)
<ipython-input-2-a91092e7ee85> in <module>()
      2 
      3 print(country_dict['Australia'])
----> 4 print(country_dict['Canada'])# This will return error

KeyError: 'Canada'

使用 get() 方法处理 KeyError

我们可以使用 get 方法来检查键。此方法接受两个参数。第一个是键，第二个是默认值。当找到键时，它将返回与该键关联的值，但是当键不存在时，它将返回默认值，该值作为第二个参数传递。

示例代码

实时演示

country_dict = {'India' : 'IN', 'Australia' : 'AU', 'Brazil' : 'BR'}
print(country_dict.get('Australia', 'Not Found'))
print(country_dict.get('Canada', 'Not Found'))

输出

AU
Not Found

使用 setdefault() 方法处理 KeyError

此 setdefault() 方法类似于 get() 方法。它也像 get() 一样接受两个参数。第一个是键，第二个是默认值。此方法的唯一区别是，当存在缺失的键时，它将使用默认值添加新的键。

示例代码

实时演示

country_dict = {'India' : 'IN', 'Australia' : 'AU', 'Brazil' : 'BR'}
country_dict.setdefault('Canada', 'Not Present') #Set a default value for Canada
print(country_dict['Australia'])
print(country_dict['Canada'])

输出

AU
Not Present

使用 defaultdict

defaultdict 是一种容器。它位于 Python 的 collections 模块中。defaultdict 以默认工厂作为其参数。最初，默认工厂设置为 0（整数）。当键不存在时，它返回默认工厂的值。

我们不需要一遍遍地指定方法，因此它为字典对象提供了更快速的方法。

示例代码

实时演示

import collections as col
#set the default factory with the string 'key not present'
country_dict = col.defaultdict(lambda: 'Key Not Present')
country_dict['India'] = 'IN'
country_dict['Australia'] = 'AU'
country_dict['Brazil'] = 'BR'
print(country_dict['Australia'])
print(country_dict['Canada'])

输出

AU
Key Not Present