使用Python字典理解进行重构

问题描述

我有2个字典，它们包含相同的键，但值对不同。让我们让dicta和dicB代表所讨论的两本词典。

dictA = {'key1':'Joe', 'key2':'Bob'}
dictB = {'key1':'Smith', 'key2':'Johnson'}

目前，我正在通过嵌套的if语句基于常见的关键字创建一个新字典。这样做时，共享一个键的值包含在新字典中的一个列表中。请参见下面的操作：

dictAB = {}  # Create a new dictionary

# Create a list container for dictionary values
for key in dictA.keys():
    dictAB[key] = []

# Iterate through keys in both dictionaries
# Find matching keys and append the respective values to the list container
for key, value in dictA.iteritems():
    for key2, value2 in dictB.iteritems():
        if key == key2:
            dictAB[key].append(value)
            dictAB[key].append(value2)
        else:
            pass

如何使用python字典理解将其转换为更干净的结构？

---

解决方案

使用集合或key views (python 2.7)：

dictAB = {k: [dictA[k], dictB[k]] for k in dictA.viewkeys() & dictB.viewkeys()}

2.7之前：

dictAB = dict((k, [dictA[k], dictB[k]]) for k in set(dictA) & set(dictB))

在python3中，您可以直接使用.keys方法进行此类操作，因为它们是以视图的形式实现的：

dictAB = {k: [dictA[k], dictB[k]] for k in dictA.keys() & dictB.keys()}

Demo(python 2.7)：

>>> dictA = {'key1':'Joe', 'key2':'Bob'}
>>> dictB = {'key1':'Smith', 'key2':'Johnson'}
>>> dictAB = {k: [dictA[k], dictB[k]] for k in dictA.viewkeys() & dictB.viewkeys()}
>>> print dictAB
{'key2': ['Bob', 'Johnson'], 'key1': ['Joe', 'Smith']}

两个集合或字典视图上的&运算符创建两个集合的交集；两个集合中存在的所有关键点。

通过使用键的交集，即使dicta或didicB的键没有出现在另一个字典中，此代码也可以正常工作。如果您绝对确定键将始终匹配，则只需直接迭代任一字典，而不使用交叉点：

dictAB = {k: [dictA[k], dictB[k]] for k in dictA}