Python代码阅读（第33篇）：反转字典

Python 代码阅读合集介绍：为什么不推荐Python初学者直接看项目源码

本篇阅读的代码实现了将一个字典进行反转，且原字典的值非唯一。

本篇阅读的代码片段来自于30-seconds-of-python。

collect_dictionary

def collect_dictionary(obj):
  inv_obj = {}
  for key, value in obj.items():
    inv_obj.setdefault(value, list()).append(key)
  return inv_obj

# EXAMPLES
ages = {
  "Peter": 10,
  "Isabel": 10,
  "Anna": 9,
}
collect_dictionary(ages) # { 10: ["Peter", "Isabel"], 9: ["Anna"] }

collect_dictionary函数接收一个字典，返回反转后的字典。函数使用dictionary.items()循环获取每一个原字典的键值对，使用dictionary.setdefault()和append()将原字典的值映射到键上。使用list()将新键的值定义为列表。

因为原字典的所有值不唯一，反转字典之后，新的键也可能对应多个值。因此反转后的字典的值是一个列表，每个列表中的元素是原字典的一个或多个键。

dictionary. setdefault(key[, default])如果字典存在键key，返回它的值。如果不存在，插入值为default的键key，并返回default。

for key, value in obj.items():
    inv_obj.setdefault(value, list()).append(key)

如果新的字典inv_obj已经有key == value，就会返回这个key对应的值（一个列表），然后使用append将原字典的键加入其中。如果新的字典inv_obj没有key == value，就会插入这个key，设置它的值为list()，并返回这个空列表。然后使用append将原字典的键加入其中。