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