思考一个小细节，从如何反转字典说起

最新推荐文章于 2022-12-01 12:41:16 发布

liu志军

最新推荐文章于 2022-12-01 12:41:16 发布

阅读量182

点赞数

本文链接：https://blog.csdn.net/lantian_123/article/details/101518477

版权

640?wx_fmt=jpeg

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里面也就不会提供这样的操作了。