关于高效使用Python字典的清单

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

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

关于高效使用Python字典的清单

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

 

  1. d = {'name''python' 
  2. if d.has_key('name'): 
  3.     pass 

good

 

  1. if 'name' in d: 
  2.     pass 

2、用 get 获取字典中的值

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

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

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

bad

 

  1. d = {'name''python'
  2. if 'name' in d: 
  3.     print(d['hello']) 
  4. else
  5.     print('default'

good

  1. print(d.get("name""default")) 

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

 

  1. data = [ 
  2.         ("animal""bear"), 
  3.         ("animal""duck"), 
  4.         ("plant""cactus"), 
  5.         ("vehicle""speed boat"), 
  6.         ("vehicle""school bus"
  7.     ] 

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

 

  1. groups = {} 
  2.  
  3. >>>  
  4. {'plant': ['cactus'],  
  5.  'animal': ['bear''duck'],  
  6.  'vehicle': ['speed boat''school bus'

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

bad

 

  1. for (key, value) in data: 
  2.     if key in groups: 
  3.         groups[key].append(value) 
  4.     else
  5.         groups[key] = [value] 

good

 

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

setdefault 的作用是:

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

4、用 defaultdict 初始化字典对象

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

 

  1. from collections import defaultdict  
  2. groups = defaultdict(list) 
  3. for (key, value) in data: 
  4.     groups[key].append(value) 

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

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

 

  1. keys = {'a''e''i''o''u' } 
  2. value = [] 
  3. d = dict.fromkeys(keys, value) 
  4. print(d) 
  5.  
  6. >>> 
  7. {'i': [], 'u': [], 'e': [],  
  8.  'a': [], 'o': []} 

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

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

 

  1. if arg == 0: 
  2.     return 'zero' 
  3. elif arg == 1: 
  4.     return 'one' 
  5. elif arg == 2: 
  6.     return "two" 
  7. else
  8.     return "nothing" 

good

 

  1. data = { 
  2.     0: "zero"
  3.     1: "one"
  4.     2: "two"
  5. data.get(arg, "nothing"

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

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

 

  1. d = { 
  2.     0: "zero"
  3.     1: "one"
  4.     2: "two"
  5.  
  6. for k, v in d.items(): 
  7.     print(k, v) 

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

 

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

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

8、使用字典推导式

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

bad

 

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

good

 

  1. numbers = [1, 2, 3] 
  2. d = {number: number * 2 for number in numbers} 
责任编辑:未丽燕 来源: FooFish的笔录
相关推荐

2018-09-28 10:45:19

Python 开发模块

2022-05-10 07:24:19

PythonCollection模块

2020-05-21 08:53:12

Python技术代码

2020-05-21 15:14:15

Python列表字典

2022-06-01 16:37:01

致态

2024-06-24 00:05:00

Python代码

2021-10-21 06:52:17

Vue3组件 API

2017-07-31 19:57:13

2010-03-15 17:26:58

Python字典

2010-05-05 14:21:37

Linux系统软件

2017-07-04 16:00:16

PythonMatplotlib可视化工具

2017-07-27 09:49:37

Python工具Matplotlib

2018-03-16 15:30:45

数据库MySQL数据字典

2018-05-22 09:07:54

数据科学语言职位

2017-05-02 21:08:35

开发架构工程师

2022-01-28 14:54:21

staticC语言编译器

2023-08-08 13:51:13

Gherkin开发

2010-03-23 16:28:31

CentOS使用

2023-09-04 19:15:19

itemPython版本

2017-05-22 15:42:39

Python字典哈希表
点赞
收藏

51CTO技术栈公众号