Python交换字典键值对的四种方法实例

时间:2022-12-05 枉费红笺人气:0

前言

在学习过程中发现有时候交换字典的键和值，会使得我们最后的输出结果更加直观明了，整理出以下四种交换方式（data是原字典，new_data 是交换后的字典）

一、当值唯一时

1. 使用zip进行交换

data= {'A':1, 'B':2, 'C':3}
new_data = dict(zip(data.values(), data.keys()))
print(new_data)
# 运行结果
# {1: 'A', 2: 'B', 3: 'C'}

2. 使用for循环遍历交换

data= {'A':1, 'B':2, 'C':3}
new_data = {}
for key, val in data.items():
    new_data[val] = key
print(new_data)
# 运行结果
# {1: 'A', 2: 'B', 3: 'C'}

3. 使用dict.items()交换

new_data = dict([val, key] for key, val in data.items())
print(new_data)
# 运行结果
# {1: 'A', 2: 'B', 3: 'C'}

上述三种方式的前提条件是值唯一。如果data = {'A': 1, 'B': 2, 'C': 3, 'D': 3}，最后输出结果是{1: 'A', 2: 'B', 3: 'D'}，C会被D代替。

二、当值不唯一时

如果值不唯一，可以使用下面这种方式，最后输出结果是{1: ['A'], 2: ['B'], 3: ['C', 'D']}

from collections import defaultdict

def main():
    data = {'A': 1, 'B': 2, 'C': 3, 'D': 3}
    new_data = defaultdict(list)
    for key, val in data.items():
        new_data[val].append(key)
    print(new_data)
# 运行结果
# {1: ['A'], 2: ['B'], 3: ['C', 'D']}

Python中通过Key访问字典，当Key不存在时，会引发‘KeyError’异常。为避免这种情况的发生，可以使用collections类中的defaultdict()方法来为字典提供默认值。该函数返回一个类似字典的对象。defaultdict是Python内建字典类（dict）的一个子类。

dict = defaultdict(factory_function)
# factory_function可以是list、set、str等等，作用是当key不存在时，返回的是工厂函数的默认值

当字典中没有的键第一次出现时，default_factory自动为其返回一个空列表，list.append()会将值添加进新列表；再次遇到相同的键时，list.append()将其值再添加进该列表。

总结

加载全部内容