用python字典理解重构

Question

我有2个字典,其中包含相同的键但值对不同.让dictA和dictB代表两个有问题的词典.

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字典理解将其变成更干净的结构？

Answer 1

使用集合或关键视图(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))

在python 3中,您可以.keys直接将此方法用于此类操作,因为它们是作为视图实现的:

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

演示(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']}

&两个集合或dict视图上的运算符创建两个集合的交集; 两个集合中存在的所有键.

通过使用键的交集,即使dictA或dictB具有未出现在其他字典中的键,此代码也将起作用.如果你绝对确定键总是匹配,你可以直接迭代任一dict而不用交集:

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