逆字典映射的性能

Question

使用value == 3获取所有dict项目并创建新dict的最有效方法是什么？

这是我到目前为止所做的：

d = {1: 2, 2: 2, 3: 1, 4: 1, 5: 1, 6: 1, 7: 1, 8: 1, ...}
new_d = {}
for item in d:
    if d[item] == 3:
        new_d[item] = d[item]

有更高效，更简单的方法吗？也许使用地图？

Answer 1

您可以使用dict comprehension：

new_d = {k:v for k, v in d.items() if v == 3}

请注意，您应该在Python 2.x中调用d.iteritems()以避免创建不必要的列表。

---

从下面的timeit.timeit测试中可以看出，此解决方案更有效：

>>> from timeit import timeit
>>> d = {1: 2, 2: 2, 3: 1, 4: 1, 5: 1, 6: 1, 7: 1, 8: 1}
>>>
>>> timeit('''
... new_d = {}
... for item in d:
...     if d[item] == 1:
...          new_d[item] = d[item]
... ''', 'from __main__ import d')
5.002458692375711
>>>
>>> timeit('new_d = {k:v for k, v in d.items() if v == 1}', 'from __main__ import d')
4.844044424640543
>>>

它也简单得多，总是很好。