Python KeyError异常处理：全面指南与实战示例

2 年 ago
新, 韵
1 minute

Python中的KeyError异常是什么？

当我们尝试从字典中访问不存在的键时，Python会引发KeyError。这是一种内置的异常类，常用于处理与字典或具有键值对的对象相关的操作。

2. Python字典中的键错误示例

让我们看一个简单的例子，程序会引发KeyError错误。

emp_dict = {'Name': 'Pankaj', 'ID': 1}

emp_id = emp_dict['ID']
print(emp_id)

emp_role = emp_dict['Role']
print(emp_role)

输出：

1
Traceback (most recent call last):
  File "/Users/scdev/Documents/PycharmProjects/hello-world/scdev/errors/keyerror_examples.py", line 6, in <module>
    emp_role = emp_dict['Role']
KeyError: 'Role'

3. 处理 Python 的KeyError 异常

我们可以使用try-except块来处理KeyError异常。下面是处理上述KeyError异常的示例。

emp_dict = {'Name': 'Pankaj', 'ID': 1}

try:
    emp_id = emp_dict['ID']
    print(emp_id)

    emp_role = emp_dict['Role']
    print(emp_role)
except KeyError as ke:
    print('员工字典中未找到键:', ke)

输出：

1
员工字典中未找到键: 'Role'

4. 在访问字典键时避免出现 KeyError

我们可以通过使用字典的get()方法来访问键值，从而避免KeyError。如果键不存在，get()方法将返回None。我们也可以在键丢失时指定一个默认值来返回。

emp_dict = {'Name': 'Pankaj', 'ID': 1}

emp_id = emp_dict.get('ID')
emp_role = emp_dict.get('Role')
emp_salary = emp_dict.get('Salary', 0)

print(f'员工信息:[ID:{emp_id}, 职位:{emp_role}, 薪资:{emp_salary}]')

输出：

员工信息:[ID:1, 职位:None, 薪资:0]

5. Pandas模块引发的KeyError

在 Pandas DataFrame 中，有些函数操作可能会引发KeyError异常，例如：

  • rename()
  • drop()

6. 参考文献

bannerAds