如何高效使用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多线程编程(四):使用Lock互斥锁
Apr 05 Python
Python单元测试框架unittest使用方法讲解
Apr 13 Python
Python更新数据库脚本两种方法及对比介绍
Jul 27 Python
Python简单实现阿拉伯数字和罗马数字的互相转换功能示例
Apr 17 Python
python合并已经存在的sheet数据到新sheet的方法
Dec 11 Python
pycharm+PyQt5+python最新开发环境配置(踩坑)
Feb 11 Python
python内存管理机制原理详解
Aug 12 Python
python 实现字符串下标的输出功能
Feb 13 Python
Python count函数使用方法实例解析
Mar 23 Python
pycharm下配置pyqt5的教程(anaconda虚拟环境下+tensorflow)
Mar 25 Python
python安装第三方库如xlrd的方法
Oct 31 Python
如何用tempfile库创建python进程中的临时文件
Jan 28 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
MySQL连接数超过限制的解决方法
2011/07/17 PHP
PHP封装的数据库保存session功能类
2016/07/11 PHP
PHP中SESSION过期设置
2021/03/09 PHP
js禁止小键盘输入数字功能代码
2011/08/01 Javascript
js 如何实现对数据库的增删改查
2012/11/23 Javascript
为jquery的ajaxfileupload增加附加参数的方法
2014/03/04 Javascript
js 通过html()及text()方法获取并设置p标签的显示值
2014/05/14 Javascript
jquery获取及设置outerhtml的方法
2015/03/09 Javascript
跟我学习javascript的this关键字
2020/05/28 Javascript
老生常谈javascript中逻辑运算符&&和||的返回值问题
2017/04/13 Javascript
Angular directive递归实现目录树结构代码实例
2017/05/05 Javascript
微信小程序 rich-text的使用方法
2017/08/04 Javascript
js实现web调用摄像头 js截取视频画面
2019/04/21 Javascript
js中arguments对象的深入理解
2019/05/14 Javascript
JS Web Flex弹性盒子模型代码实例
2020/03/10 Javascript
Javascript实现贪吃蛇小游戏(含详细注释)
2020/10/23 Javascript
[01:05:00]2018国际邀请赛 表演赛 Pain vs OpenAI
2018/08/24 DOTA
python使用cPickle模块序列化实例
2014/09/25 Python
python rsa 加密解密
2017/03/20 Python
python批量修改图片后缀的方法(png到jpg)
2018/10/25 Python
Python中的 enum 模块源码详析
2019/01/09 Python
Python实现 PS 图像调整中的亮度调整
2019/06/28 Python
django 数据库连接模块解析及简单长连接改造方法
2019/08/29 Python
python库matplotlib绘制坐标图
2019/10/18 Python
Matplotlib绘制雷达图和三维图的示例代码
2020/01/07 Python
python文件操作seek()偏移量,读取指正到指定位置操作
2020/07/05 Python
eDreams葡萄牙:全球最大的在线旅行社之一
2019/04/15 全球购物
英国折扣高尔夫商店:Discount Golf Store
2019/11/19 全球购物
个人思想理论学习的自我鉴定
2013/11/30 职场文书
采购主管的岗位职责
2013/12/17 职场文书
安全责任协议书
2014/04/21 职场文书
实践单位评语
2014/04/26 职场文书
2014年仓库工作总结
2014/11/20 职场文书
简爱读书笔记
2015/06/26 职场文书
服务行业标语口号
2015/12/26 职场文书
pd.DataFrame中的几种索引变换的实现
2022/06/16 Python