如何高效使用Python字典的方法详解


Posted in Python onAugust 31, 2017

前言

众所周知字典(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 是最好的选择。

bad

d = {'name': 'python'}

if d.has_key('name'):
 pass

good

if '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

bad

d = {'name': 'python'}
if 'name' in d:
 print(d['hello'])
else:
 print('default')

good

print(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 方法。

bad

for (key, value) in data:
 if key in groups:
  groups[key].append(value)
 else:
  groups[key] = [value]

good

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

setdefault 的作用是:

  • 如果 key 存在于字典中,那么直接返回对应的值,等效于 get 方法
  • 如果 key 不存在字典中,则会用 setdefault 中的第二个参数作为该 key 的值,再返回该值。

4、用 defaultdict 初始化字典对象

如果不希望 d[x] 在 x 不存在时报错,除了在获取元素时使用 get 方法之外,另外一种方式是用 collections 模块中的 defaultdict,在初始化字典的时候指定一个函数,其实 defaultdict 是 dict 的子类。

from collections import defaultdict

groups = defaultdict(list)
for (key, value) in data:
 groups[key].append(value)

当 key 不存在于字典中时,list 函数将被调用并返回一个空列表赋值给 d[key],这样一来,你就不用担心调用 d[k] 会报错了。

5、用 fromkeys 将列表转换成字典

keys = {'a', 'e', 'i', 'o', 'u' }
value = []
d = dict.fromkeys(keys, value)
print(d)

>>>
{'i': [], 'u': [], 'e': [], 
 'a': [], 'o': []}

6、用字典实现 switch ... case 语句

Python 中没有 switch ... case 语句,这个问题Python之父龟叔表示这个语法过去没有,现在没有,以后也不会有。因为Python简洁的语法完全可以用 if ... elif 实现。如果有太多的分支判断,还可以使用字典来代替。

if arg == 0:
 return 'zero'
elif arg == 1:
 return 'one'
elif arg == 2:
 return "two"
else:
 return "nothing"

good

data = {
 0: "zero",
 1: "one",
 2: "two",
}
data.get(arg, "nothing")

7、使用 iteritems 迭代字典中的元素

python提供了几种方式迭代字典中的元素,第一种是使用 items 方法:

d = {
 0: "zero",
 1: "one",
 2: "two",
}

for k, v in d.items():
 print(k, v)

items 方法返回的是(key ,value)组成的列表对象,这种方式的弊端是迭代超大字典的时候,内存瞬间会扩大两倍,因为列表对象会一次性把所有元素加载到内存,更好的方式是使用 iteritems

for k, v in d.iteritems():
 print(k, v)

iteritems 返回的是迭代器对象,迭代器对象具有惰性加载的特性,只有真正需要的时候才生成值,这种方式在迭代过程中不需要额外的内存来装载这些数据。注意 Python3 中,只有 items 方法了,它等价于 Python2 中的 iteritems,而 iteritems 这个方法名被移除了。

8、使用字典推导式

推导式是个绝妙的东西,列表推导式一出,map、filter等函数黯然失色,自 Python2.7以后的版本,此特性扩展到了字典和集合身上,构建字典对象无需调用 dict 方法。

bad

numbers = [1,2,3]
d = dict([(number,number*2) for number in numbers])

good

numbers = [1, 2, 3]
d = {number: number * 2 for number in numbers}

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作能带来一定的帮助,如果有疑问大家可以留言交流,谢谢大家对三水点靠木的支持。

Python 相关文章推荐
python查看zip包中文件及大小的方法
Jul 09 Python
python中reduce()函数的使用方法示例
Sep 29 Python
简单实现python进度条脚本
Dec 18 Python
Python实现pdf文档转txt的方法示例
Jan 19 Python
Python数据持久化shelve模块用法分析
Jun 29 Python
python 不同方式读取文件速度不同的实例
Nov 09 Python
python实现微信定时每天和女友发送消息
Apr 29 Python
20行python代码实现人脸识别
May 05 Python
matplotlib jupyter notebook 图像可视化 plt show操作
Apr 24 Python
Python  word实现读取及导出代码解析
Jul 09 Python
Django集成MongoDB实现过程解析
Dec 01 Python
python二维图制作的实例代码
Dec 03 Python
利用Python爬取微博数据生成词云图片实例代码
Aug 31 #Python
Python利用ElementTree模块处理XML的方法详解
Aug 31 #Python
go和python变量赋值遇到的一个问题
Aug 31 #Python
理解Python中的绝对路径和相对路径
Aug 30 #Python
python 递归遍历文件夹,并打印满足条件的文件路径实例
Aug 30 #Python
python递归打印某个目录的内容(实例讲解)
Aug 30 #Python
python-opencv在有噪音的情况下提取图像的轮廓实例
Aug 30 #Python
You might like
火车采集器 免费版使出收费版本功能实现原理
2009/09/17 PHP
php后门URL的防范
2013/11/12 PHP
PHP魔术方法之__call与__callStatic使用方法
2017/07/23 PHP
Docker搭建自己的PHP开发环境
2018/02/24 PHP
Asp.net下利用Jquery Ajax实现用户注册检测(验证用户名是否存)
2010/09/12 Javascript
JS俄罗斯方块,包含完整的设计理念
2010/12/11 Javascript
JSON.stringify 语法实例讲解
2012/03/14 Javascript
jquery判断元素的子元素是否存在的示例代码
2014/02/04 Javascript
js实现的黑背景灰色二级导航菜单效果代码
2015/08/24 Javascript
JS实现仿腾讯微博无刷新删除微博效果代码
2015/10/16 Javascript
javascript和jQuery实现网页实时聊天的ajax长轮询
2016/07/20 Javascript
关于JavaScript和jQuery的类型判断详解
2016/10/08 Javascript
JavaScript数组的5种迭代方法
2017/09/29 Javascript
jQuery除指定区域外点击任何地方隐藏DIV功能
2017/11/13 jQuery
vue实现自定义日期组件功能的实例代码
2018/11/06 Javascript
基于Vue实现平滑过渡的拖拽排序功能
2019/06/12 Javascript
[01:32:22]DOTA2-DPC中国联赛 正赛 Ehome vs VG BO3 第一场 2月5日
2021/03/11 DOTA
编写Python爬虫抓取豆瓣电影TOP100及用户头像的方法
2016/01/20 Python
python探索之BaseHTTPServer-实现Web服务器介绍
2017/10/28 Python
Python实现查看系统启动项功能示例
2018/05/10 Python
Python IDLE清空窗口的实例
2018/06/25 Python
详解Python3定时器任务代码
2019/09/23 Python
使用pandas库对csv文件进行筛选保存
2020/05/25 Python
Python 如何创建一个线程池
2020/07/28 Python
草莓网中国:StrawberryNet中国
2020/08/17 全球购物
小学生防溺水广播稿
2014/01/12 职场文书
公务员综合考察材料
2014/02/01 职场文书
《九色鹿》教学反思
2014/02/27 职场文书
充分就业社区汇报材料
2014/05/07 职场文书
增员口号大全
2014/06/18 职场文书
公司开会通知
2015/04/20 职场文书
《落花生》教学反思
2016/02/16 职场文书
Java用自带的Image IO给图片添加水印
2021/06/15 Java/Android
CSS3实现列表无限滚动/轮播效果
2021/06/23 HTML / CSS
Java后台生成图片的完整步骤
2021/08/04 Java/Android
使用CSS实现按钮边缘跑马灯动画
2023/05/07 HTML / CSS