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将原字典的键加入其中。

上一篇:print 函数改造


下一篇:计划任务