如何交叉字典？ [复制]

Question

我已经很久没有使用 python 并且有点生疏了，但是对于第一个问题，我正在使用字典并且需要将其相交以返回键和值。所以例如，我正在输入

a = {1:'a1', 2.5:'a2', 4:'a3'}
b = {1:'a1', 3:'a2',  5:'a4'}

如果我输入c = intersect(a,b)，我希望它返回{1:'a1'}，但我只返回{'a1'}。

到目前为止我的代码是：

def intersect(a, b):
    for i in a:
        if j in b:
            if a[i]==b[i]:
                return ({i})
        else:
            return {}

Answer 1

这是 intersect 函数的一个版本，它与键相交并使用第一个字典的值，这意味着在不同值的情况下，此函数会偏向于第一个字典中的值：

def intersect(a, b):
    return {k: a[k] for k in a.keys() & b.keys()}

这是可行的，因为dict.keys()（自 Python 3 起）返回的 dict_keys 视图对象是集合（即实现 collections.abc.Set）。

如果你想自定义，使用哪个值，我们可以使用一个函数，它接受两个值：

def intersect(a, b, select=lambda a, b: a):
    return {k: select(a[k], b[k]) for k in a.keys() & b.keys()}

除了返回其中一个值之外，我们还可以例如添加值：

>>> intersect({'a': 4, 'b': 3}, {'a': 5}, lambda a, b: a + b)
{'a': 9}

Answer 2

您可以使用 actual set 交叉点来简化此操作。这仅保留共享项（键和值相同）：

def intersect(a, b):
    return dict(a.items() & b.items())

如果您的 dict 参数具有不可散列的值，您可以按照以下几行进行 dict 理解：

def intersect(a, b):
    return {k: a[k] for k in a if b.get(k, not a[k]) == a[k]}

你可以像这样修复你原来的方法：

def intersect(a, b):
    d = {}
    for i in a:
        if i in b and a[i] == b[i]:
            d[i] = a[i]
    return d