亲宝软件园·资讯

展开

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

枉费红笺 人气: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()将其值再添加进该列表。

总结

加载全部内容

相关教程
猜你喜欢
用户评论