2025年3月28日 星期五 甲辰(龙)年 月廿七 设为首页 加入收藏
rss
您当前的位置:首页 > 计算机 > 编程开发 > Python

Python:操作dict时避免出现KeyError的几种方法

时间:10-23来源:作者:点击数:35

在读取dictkeyvalue时,如果key不存在,就会触发KeyError错误,如:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • print(t['d'])

就会出现:

  • KeyError: 'd'

第一种解决方法

首先测试key是否存在,然后才进行下一步操作,如:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • if 'd' in t:
  • print(t['d'])
  • else:
  • print('not exist')

会出现:

  • not exist

第二种解决方法

利用dict内置的get(key[,default])方法,如果key存在,则返回其value,否则返回default;使用这个方法永远不会触发KeyError,如:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • print(t.get('d'))

会出现:

  • None

加上default参数:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • print(t.get('d', 'not exist'))
  • print(t)

会出现:

  • not exist
  • {'a': '1', 'c': '3', 'b': '2'}

第三种解决方法

利用dict内置的setdefault(key[,default])方法,如果key存在,则返回其value;否则插入此key,其valuedefault,并返回default;使用这个方法也永远不会触发KeyError,如:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • print(t.setdefault('d'))
  • print(t)

会出现:

  • None
  • {'b': '2', 'd': None, 'a': '1', 'c': '3'}

加上default参数:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • print(t.setdefault('d', 'not exist'))
  • print(t)

会出现:

  • not exist
  • {'c': '3', 'd': 'not exist', 'a': '1', 'b': '2'}

第四种解决方法

向类dict增加__missing__()方法,当key不存在时,会转向__missing__()方法处理,而不触发KeyError,如:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • class Counter(dict):
  • def __missing__(self, key):
  • return None
  • c = Counter(t)
  • print(c['d'])

会出现:

  • None

更改return值:

  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • class Counter(dict):
  • def __missing__(self, key):
  • return key
  • c = Counter(t)
  • print(c['d'])
  • print(c)

会出现:

  • d
  • {'c': '3', 'a': '1', 'b': '2'}

第五种解决方法

利用collections.defaultdict([default_factory[,...]])对象,实际上这个是继承自dict,而且实际也是用到的__missing__()方法,其default_factory参数就是向__missing__()方法传递的,不过使用起来更加顺手:

如果default_factoryNone,则与dict无区别,会触发KeyError错误,如:

  • import collections
  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • t = collections.defaultdict(None, t)
  • print(t['d'])

会出现:

  • KeyError: 'd'

但如果真的想返回None也不是没有办法:

  • import collections
  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • def handle():
  • return None
  • t = collections.defaultdict(handle, t)
  • print(t['d'])

会出现:

  • None

如果default_factory参数是某种数据类型,则会返回其默认值,如:

  • import collections
  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • t = collections.defaultdict(int, t)
  • print(t['d'])

会出现:

  • 0

又如:

  • import collections
  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • t = collections.defaultdict(list, t)
  • print(t['d'])

会出现:

  • []

注意:

如果dict内又含有dictkey嵌套获取value时,如果中间某个key不存在,则上述方法均失效,一定会触发KeyError

  • import collections
  • t = {
  • 'a': '1',
  • 'b': '2',
  • 'c': '3',
  • }
  • t = collections.defaultdict(dict, t)
  • print(t['d']['y'])

会出现:

  • KeyError: 'y'
方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门