首页 > 编程 > Python > 正文

高效使用Python字典的清单

2020-02-22 23:35:31
字体:
来源:转载
供稿:网友

字典(dict)对象是 Python 最常用的数据结构,社区曾有人开玩笑地说:"Python企图用字典装载整个世界",字典在Python中的重要性不言而喻,这里整理了几个关于高效使用字典的清单,希望Python开发者可以在日常应用开发中合理利用,让代码更加 Pythonic。

1、用 in 关键字检查 key 是否存在

Python之禅中有一条开发哲学是:

There should be one-- and preferably only one --obvious way to do it.

尽量找一种,最好是唯一种显而易见的解决方案。Python2 中判断某个 key 是否存在字典中可使用 has_key 方法,另外一种方式是使用 in 关键字。但是强烈推荐使用后者,因为 in 的处理速度更快,另外一个原因是 has_key 这个方法在 Python3 被移除了,要想同时兼容py2和py3两个版本的代码,用 in 是最好的选择。

badd = {'name': 'python'}if d.has_key('name'):  passgoodif 'name' in d:  pass

2、用 get 获取字典中的值

关于获取字典中的值,一种简单的方式就是用d[x]访问该元素,但是这种情况在 key 不存在的情况下会报 KeyError 错误,当然你可以先用 in 操作检查 key 是否在字典中再获取,不过这种方式不符合 Python 之禅中说的:

Simple is better than complex.
Flat is better than nested.

好的代码应该是简单易懂的,扁平的代码结构更加可读。我们可以使用 get 方法来代替 if ... else

badd = {'name': 'python'}if 'name' in d:  print(d['hello'])else:  print('default')goodprint(d.get("name", "default"))

3、用 setdefault 为字典中不存在的 key 设置缺省值

data = [    ("animal", "bear"),    ("animal", "duck"),    ("plant", "cactus"),    ("vehicle", "speed boat"),    ("vehicle", "school bus")  ]

在做分类统计时,希望把同一类型的数据归到字典中的某种类型中,比如上面代码,把相同类型的事物用列表的形式重新组装,得到新的字典

groups = {}>>> {'plant': ['cactus'],  'animal': ['bear', 'duck'],  'vehicle': ['speed boat', 'school bus']}

普通的方式就是先判断 key 是否已经存在,如果不存在则要先用列表对象进行初始化,再执行后续操作。而更好的方式就是使用字典中的 setdefault 方法。

badfor (key, value) in data:  if key in groups:    groups[key].append(value)  else:    groups[key] = [value]goodgroups = {}for (key, value) in data:  groups.setdefault(key, []).append(value)

setdefault 的作用是:

如果 key 存在于字典中,那么直接返回对应的值,等效于 get 方法

如果 key 不存在字典中,则会用 setdefault 中的第二个参数作为该 key 的值,再返回该值。

4、用 defaultdict 初始化字典对象

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表