• 对Python的思考:从如何反转字典说起
  • 发布于 2个月前
  • 351 热度
    1 评论
在某些特定的场景,你会遇到这样的需求,将字典的 key 和 value 进行互换,例如:
d = { 'a': 1, 'b':2}
变成
d = { 1: 'a', 2: 'b' }
你可能最先能想到的土办法就是遍历字典,逐个将 key-value 互换成 value-key 放进新字典中
d = {'a': 1, 'b': 2}
new_d = dict()
for key in d:
    value = d[key]
    new_d[value] = key
其实啊,如果你知道推导式概念的话,那么用一行代码就可以搞定。自从 Python2.7和 Python3 加入了列表推导式、集合推导式和字典推导式之后,解决这类问题就得心应手了。用字典推导式可优雅写出:

new_d = {v: k for k, v in d.items()}
如果你熟悉列表推导式,上面的代码你也觉得很眼熟,就是将 key,value 互换构成新的字典。也有人可能会推荐你使用函数式编程风格的代码来实现:
>>> d = { 'a': 1, 'b':2 }
>>> dict((map(reversed, d.items())))
{1: 'a', 2: 'b'}
函数式编程风格的代码看起来可读性并没有那么好,而且效率上也比不上推导式,我们可以用 timeit 证实一下
>>> import timeit
>>> timeit.timeit("dict((map(reversed, {'a': 1, 'b': 2}.items())))", number=100000)
0.16313515009824187
>>> timeit.timeit("{v: k for k, v in  {'a': 1, 'b': 2}.items()}", number=100000)
0.06065327790565789
一看速度上慢将近 3 倍,自从 Python  引入了推导式特性后, map、filter、reduce 这样的函数就很少被推荐使用了。

在来思考一个问题,为什么字典自己没有提供这样的 API 接口给我们使用?

根据字典的特性:

字典的 key 必须是唯一的,相同的 key 会被覆盖, 而 value 可以相同,所以这个反转,就是 key 和 value 都是唯一的场景下。

key 必须是可 hash 的,所以如果 value 是不可 hash 的对象,比如列表,就不能做为 key 了。

因此字典反转只适合某些特定场景,而不适合当做一个通用操作来使用,所以字典的API里面也就不会提供这样的操作了。
用户评论