处理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()Method处理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()Method处理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