有网友问:
用Python反转字典时,如遇到多个键对应的值相同,该如何将反转后的这些键并为一个列表?修改
比如说{'a':1,'b':1,'c':2}反转为{1:['a','b'],2:'c'}
解题思路:
1、遍历字典、交换key 和value。
2、如果key 相同,那么值为list,否则值为一个字符串。
下面看代码吧。
"""黄哥Python培训 黄哥所写"""def reverse_dict(d): res = {} if len(d) == 0: return res for key, val in d.items(): if val not in res: res[val] = key else: old_val = res[val] if isinstance(old_val, list): old_val.append(key) res[val] = old_val else: res[val] = [old_val, key] return resif __name__ == '__main__': d = {} print(reverse_dict(d)) d = {'a': 1, 'b': 1, 'c': 2} print(reverse_dict(d)) d = {'a': 1, 'b': 1, 'c': 1, 'd': 2, 'e': 2, 'f': 3} print(reverse_dict(d))